35-搜索插入位置(Go)

1 阅读1分钟

题目

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

请必须使用时间复杂度为 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

我的代码

对于顺序数组,且时间复杂度为 O(log n) ,二分法是一个不错的方法。

func searchInsert(nums []int, target int) int {
    left, right := 0, len(nums)-1
    if target < nums[left]{
        return 0;
    }
    if target > nums[right]{
        return len(nums)
    }
    for left<=right {
        mid := (left+right)/2
        if nums[mid]== target {
           return mid
        } else  if nums[mid] < target {
             // 注意这里的加一
             left = mid+1
        } else {
            // 注意这里的减一
            right =mid-1
        }
    }
    // 注意这里是left
    return left
}