题目描述

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

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

提示:

• 1 <= nums.length <= 104
• -104 <= nums[i] <= 104
• nums 为 无重复元素 的 升序 排列数组
• -104 <= target <= 104

解题思路

• 可以循环枚举，找到第一个大于等于的位置即可，O(n)

func searchInsert(nums []int, target int) int {
   ans := 0
   for ans < len(nums) && nums[ans] < target {
      ans++
   }
   return ans
}

• O(logn)级别用二分查找即可

func searchInsert2(nums []int, target int) int {
   l, r, mid := 0, len(nums)-1, 0
   for l < r {
      mid = (l + r) / 2
      if nums[mid] < target {
         l = mid + 1
      } else {
         r = mid
      }
   }
   if nums[l] < target {
      return l + 1
   }
   return l
}