python二分查找

Posted 2021-03-01 刘通1997

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

你可以假设数组中无重复元素。

示例 1:

输入: [1,3,5,6], 5
输出: 2
示例 2:

输入: [1,3,5,6], 2
输出: 1
示例 3:

输入: [1,3,5,6], 7
输出: 4
示例 4:

输入: [1,3,5,6], 0
输出: 0

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        r=len(nums)-1
        l=0
        if target<nums[0]:
            return 0
        if target>nums[r]:
            return r+1
        ans =0
        while l<=r:
            mid=(l+r)//2
            if target==nums[mid]:
                return mid
            elif target>nums[mid]:
                ans = mid + 1
                l = mid + 1
            elif target<nums[mid]:
                r=mid-1
        return ans