[LeetCode][golang] 35. 搜索插入位置

Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。

题目描述：

35. 搜索插入位置

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

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

示例 4:

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

示例 5:

输入: nums = [1], target = 0
输出: 0

提示:

1 <= nums.length <= 10 的 4次方
-10 的 4次方 <= nums[i] <= 10 的 4次方
nums 为无重复元素的升序排列数组
-10 的 4次方 <= target <= 10 的 4次方
通过次数710,006提交次数1,561,369

思路分析：

题目的前提是无重复元素的升序排列数组，所以可使用二分查找法。
如果在数组中找到目标值，则返回目标值的下标。
如果没有找到目标值，则返回二分查找最后结束的位置即可。

AC 代码：

golang ：

// 搜索插入位置
func SearchInsert(nums []int, target int) int {
   low := 0
   high := len(nums) - 1

   n := -1
   for low <= high {
      // 中间位置等于 低位位置 + （高位位置-低位位置）/2
      mid := low + (high-low)/2
      
      // 找到目标值，记录位置，跳出循环
      if target == nums[mid] {
         n = mid
         break
      }

      // 当前值小于目标值，则低位位置 改为 中间位置 + 1
      if target < nums[mid] {
         high = mid - 1
      }
      // 当前值大于目标值，则高位位置 改为 中间位置 - 1
      if target > nums[mid] {
         low = mid + 1
      }
   }

   // 未找到目标值，则应在上面循环结束的位置添加目标值
   if n == -1 {
      n = low
   }

   return n
}

总结：

1. 所有无重复元素的有序数组的处理，都可以使用二分查找法。

2. 找不到目标值时，可以进行插入处理。

---