力扣HOT100(36)二分查找-搜索插入位置

题目核心要求

给定一个无重复元素的升序数组和一个目标值:

  1. 如果目标值存在,返回它的索引
  2. 如果目标值不存在,返回它按顺序应该插入的位置
  3. 必须用 O (log n) 时间复杂度(也就是只能用二分)

这道题的精髓就是把两个要求合并成一个问题

求:数组中第一个大于等于 target 的元素的下标

为什么?

  • 如果 target 存在:第一个大于等于 target 的元素就是 target 本身,返回它的索引
  • 如果 target 不存在:第一个大于等于 target 的元素的位置,就是 target 应该插入的位置
  • 如果 target 比所有数都大:没有大于等于 target 的元素,插入位置就是数组长度

这一步转化是这道题的核心,理解了这个,代码就水到渠成了。

和普通二分查找的区别

普通二分(找等于 target) 本题二分(找第一个 >=target)
找到直接返回 mid 找到不直接返回,继续往左找更左边的
没找到返回 - 1 没找到返回插入位置
不需要额外变量存结果 需要 ans 变量存最终结果
cpp 复制代码
class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int n  = nums.size();
        int left = 0,right = n-1,ans = n;//如果target比所有的数都大 那么直接返回ans = n;
        while(left <= right){//当左小于等于右的时候
            int mid = ((right - left)/2) + left;//不写成right + left /2的形式 是因为避免left和right太大 导致栈溢出
            if(target <= nums[mid]){
                ans = mid;//缩小区域,如果下面还是都比target小 那么就返回mid
                right = mid - 1;//缩短到0到mid-1

            }
            else{
                left = mid +1 ;//如果target比中间值大 那么就把左边界右移
            }
            
        }
        return ans;
    }
};
  • 先留好最坏情况的退路 :把 ans 初始化为数组长度 n,提前处理「target 比所有数都大,插在最后」的情况
  • 不断缩小搜索范围
    • 如果 target <= nums[mid]:说明 mid 是一个可能的插入位置 ,先记下来;然后往左缩,看看有没有更靠前的位置
    • 如果 target > nums[mid]:说明 mid 左边肯定插不了,往右缩,去右边找
  • 循环结束,答案自然出来 :当区间空了(left > right),ans 里存的就是第一个大于等于 target 的位置,也就是最终的插入位置
相关推荐
kkkkkkkkkk_Z5 小时前
学嵌入式和C语言编程数据结构|学习日记Day22:栈,二叉树,哈希存储
c语言·数据结构·学习
Navigator_Z5 小时前
LeetCode //C - 1206. Design Skiplist
c语言·算法·leetcode
码行山野赴时序归途5 小时前
三道经典数组题:从暴力到最优的算法思维
c语言·开发语言·数据结构·算法·leetcode
徐小夕5 小时前
3分钟从想法到Agent上线:我们开源了一款AI可视化工作流“IDE”
前端·算法·github
sel_96 小时前
【强化学习】Hands-on Modern RL项目实践|OPD 算法完整解析
人工智能·深度学习·算法·机器学习·语言模型
Navigator_Z7 小时前
LeetCode //C - 1209. Remove All Adjacent Duplicates in String II
c语言·算法·leetcode
ShineWinsu9 小时前
对于 C++:C++14中从变量模板、泛型 Lambda 到并发与字面量的解析
c++·算法
土司大王9 小时前
LeetCode hot100——合并两个有序链表
算法·leetcode·链表
码行山野赴时序归途9 小时前
算法效率的度量(上):时间复杂度中的对数思维
c语言·数据结构·算法