二分查找#

35. 搜索插入位置

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

请必须使用时间复杂度为 O(log n) 的算法。

---

示例 1:

1
输入: nums = [1,3,5,6], target = 5
2
输出: 2

示例 2:

1
输入: nums = [1,3,5,6], target = 2
2
输出: 1

示例 3:

1
输入: nums = [1,3,5,6], target = 7
2
输出: 4

---

思路

这题我感觉唯一需要思考的就是不存在时的插入位置

1. 数组中插入位置有三种可能：最左侧和数列中间和最右侧，最左侧和最右侧很容易推倒出来可以return left解决，就需要考虑数列中间的情况
2. 由于left <= right和mid+1，所以最后left和right和mid必然重合，且mid一定在他的左侧或右侧，距离为1
3. 所以最后无论是target < mid导致right - 1，还是target > mid导致left + 1，left必然是target的插入位置，所以最后return left即可

1
func searchInsert(nums []int, target int) int {
2
    left, right := 0, len(nums)-1
3
    mid := 0
4
    for left <= right {
5
        mid = left + (right - left)/2
6

7
        if nums[mid] == target {
8
            return mid
9
        }else if nums[mid] < target {
10
            left = mid + 1
11
        }else {
12
            right = mid - 1
13
        }
14
    }
15
    return left
16
}