给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
解题思路
二分查找插入,指定左右指针,要找到一个位置ans,使得target==ans的值或者 nums[ans-1]<target<nums[ans]
将target与mid的值对比,如果相等,则返回mid的值,如果target比mid的值大,left值变为mid+1,继续;如果target比mid的值小,那么target肯定是插在mid或者mid的左边,记录此时mid的值,right变为mid-1,继续遍历,如果后面没有比target大的值,如果没有比target的值了,那么插入的位置就是当前的值,否则继续变化ans的值
代码
class Solution(object):
def searchInsert(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
if target < nums[0]:
return 0
if target > nums[-1]:
return len(nums)
left = 0
right = len(nums) - 1
ans=len(nums)
while left <= right:
mid = (right - left) / 2 + left
if nums[mid] == target:
ans = mid
return ans
if nums[mid] > target:
ans = mid
right = mid-1
else:
left = mid+1
return ans