Leetcode解题之路(golang版)：35. 搜索插入位置(Search Insert Position)

35. 搜索插入位置

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

你可以假设数组中无重复元素。

示例 1:
输入: [1,3,5,6], 5
输出: 2

示例 2:
输入: [1,3,5,6], 2
输出: 1

示例 3:
输入: [1,3,5,6], 7
输出: 4

示例 4:
输入: [1,3,5,6], 0
输出: 0

如果我告诉你们leetcode显示的耗时，暴力查找比二分查找耗时短，你们会怎么想…

解法一：

空间复杂度：$O(1)$

时间复杂度：$O(n)$

//暴力解决问题，顺序查找
func searchInsert(nums []int, target int) int {
    for i := 0; i < len(nums); i++ {
        if nums[i] >= target {
            return i
        } else {
            if i == len(nums) - 1 {
                return i + 1
            }
        }
    }
    return len(nums) - 1
}

解法二：

空间复杂度：$O(1)$

时间复杂度：$O(logN)$

//二分法
//可以想到这是一道类似于在0 0 0 0 1 1 1 1数列中找到第一个1的问题
//那么0和1的意义要明确清楚
//0代表target大于当前值
//1代表target小于等于当前值
//但是还有一种特殊情况我们需要判断
//当target大于所有的当前值时
//left会越界，所以在循环条件中要加上left < len(nums)
func searchInsert(nums []int, target int) int {
    left := 0
    right := len(nums)
    mid := 0
    for left < right && left < len(nums) {
        mid = (left + right) >> 1
        if target > nums[mid] {
            left = mid + 1
        } else {
            right = mid
        }
    }
    return left
}