Skip to content

【力扣】35. 搜索插入位置 学习大神的二分法查找

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

春招打卡第20天第28篇。

勤学似春起之苗,不见其增,日有所长;辍学如磨刀之石,不见其损,日有所亏。

掘金的活动真多哇,这个月决定每天用go刷题,一方面提升一下算法水平,另一方面沉淀一下go语言的学习。

Let's GO!

题目描述

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

请必须使用时间复杂度为 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 <= 10410^{4}104

−104-10^{4}−104 <= nums[i] <= 10410^{4}104

nums 为无重复元素的升序排列数组

−104-10^{4}−104 <= target <= 10410^{4}104

题目分析

看完题目,第一反应就是用二分查找法解题。

但是又有点纠结,因为题目还有个条件是:如果目标值不存在于数组中,返回它将会被按顺序插入的位置。

这个要怎么搞才好呢?

好好想了一下,二分法是可以解题的,因为如果目标值不存在数组中,那么他应该被插入的位置就是找到比他小的数字的下标+1。

我又参考了一下大神的解题思路:

不断用二分法逼近查找第一个大于等于 target 的下标 。

思路讲解

  1. 我们获得数组的长度,使用二分法进行查找
  2. ans 初值设置为数组长度可以省略边界条件的判断(大神不愧是大神,开始我就没考虑到~)

AC代码

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

运行结果

image.png

总结

复杂度分析

时间复杂度:O(logn),其中 n 为数组的长度。

空间复杂度:O(1)。我们只需要常数空间存放若干变量。

来源说明

来源:力扣(LeetCode)

链接:leetcode-cn.com/problems/se…

著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

最后

感谢阅读,欢迎大家三连:点赞、收藏、投币(关注)!!!

8e95dac1fd0b2b1ff51c08757667c47a.gif

🚀 学习遇到瓶颈?想进大厂?

看完这篇技术文章,如果还是觉得不够系统,或者想在实战中快速提升?
王中阳的就业陪跑训练营,提供定制化学习路线 + 企业级实战项目 + 简历优化 + 模拟面试。

了解训练营详情