给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
示例 1:
输入: nums = [1,3,5,6], target = 5
输出: 2
示例 2:
输入: nums = [1,3,5,6], target = 2
输出: 1
示例 3:
输入: nums = [1,3,5,6], target = 7
输出: 4
使用二分法就可以很快解决了,需要判断的是究竟return哪个?
if target == nums[mid]
这种情况就可以直接return mid了
另外的条件判断target > nums[mid] 和 target < nums[mid]的情况
1.当target在左区间,我们需要right-1,举个例子1,3,5,7,9 mid=2 nums[2]=5,我们需要插入target=4
right=mid-1 所以right=1,那么这种情况我们需要最后return right+1 就是target需要插入的位置.
2.当target在右区间,我们需要插入target=6 nums[2]=5, target在左区间, left=mid+1=3,right=4
mid=(3+4)/2=3, nums[3]=7>target=6, target又回到了左区间,right=mid-1=2 right<left,结束while。
这个时候就知道6放在5后面, return right+1
3.可以跑一次放在第一位和最后一位也是right+1
4.所以最后return right+1即可
func searchInsert(nums []int, target int) int {
left,right:=0,len(nums)-1
for left<=right {
mid:=left+(right-left)>>1
if target == nums[mid]{
return mid
} else if target > nums[mid] {
left=mid+1
} else {
right=mid-1
}
}
// 分别处理如下四种情况
// 目标值在数组所有元素之前 [0, -1]
// 目标值等于数组中某一个元素 return middle;
// 目标值插入数组中的位置 [left, right],return right + 1
// 目标值在数组所有元素之后的情况 [left, right], 因为是右闭区间,所以 return right + 1
return right+1
}