刷题使我快乐,满脸开心.jpg

  • 来源:力扣(LeetCode)
  • 链接:https://leetcode.cn/problems/search-insert-position
  • 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目

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

请必须使用时间复杂度为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(log n),那基本就是二分法或类似的思路,递归或者回溯就是比较好想到的办法,而递归回溯的关键点就在于循环条件和边界点。

循环条件

对于这个问题来说,循环条件很简单,就是大于还是小于中值,等于划在哪边均可,只是边界点可能会有所区别。

边界点

至于边界点,最后一定是走到了元素个数就剩下了一个或者两个这种已经可以判断出准确位置的地方,只需要看目标值跟仅剩的值的大小关系即可了,代码已经呼之欲出了

代码

func searchInsert(nums []int, target int) int {
	sliceLen := len(nums)
	mid := sliceLen / 2
	if sliceLen <= 2 {
		if nums[mid] < target {
			return sliceLen
		}
		if nums[0] >= target {
			return 0
		}
		return mid
	}

	if nums[mid] > target {
		return searchInsert(nums[:mid], target)
	} else {
		return mid + searchInsert(nums[mid:sliceLen], target)
	}
}