leetcode35 搜索插入位置

题目

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

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

示例

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

解析

这道题和之前的区别就是，在找不到的时候，返回需要被插入的位置，这个位置其实就是数组中第一个大于target的下标：

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

找不到的情况下，返回大于target的第一个下标即可，也就是left。
这道题的关键点在于，数组中没有target的时候，怎么确定返回的就是大于target的第一个值。以本题举例，target=2, 最后一次循环的时候，left=0，right=1（都指的是下标），mid算出来是0，nums[0]首先!=target，然后满足nums[mid] < target，则left=mid+1就等于right了。
在这个左闭右开写法的代码中，跳出循环的条件就是left =right，等于的时候，返回left或right都可以