力扣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 的位置,也就是最终的插入位置
相关推荐
alexwang21113 分钟前
HDU 4348 详细题解
c++·算法·题解·hdu·主席树·可持久化数据结构·可持久化线段树
可编程芯片开发1 小时前
基于分段变步长搜索RMDCFT算法的天基雷达空间机动目标检测方法MATLAB仿真,对比RMDCFT算法
算法
2501_946736161 小时前
在线考试平台如何选型?从功能、优势与应用场景角度详解
大数据·人工智能·算法
天辛大师1 小时前
天心大师:不确定中锚定自我,AI生活的哲学命题
人工智能·算法·决策树·机器学习·生活·启发式算法
皓月斯语2 小时前
B2132 素数对
c++·算法·题解
星轨初途3 小时前
LeetCode 热题 100——day2 字母异位词分组
c++·算法·leetcode
happyprince3 小时前
篇1:整体观 · bitsandbytes项目全貌解读
人工智能·算法
Livia要学习3 小时前
Python2和Python3字典底层原理
数据结构
青梅橘子皮3 小时前
STL---map/set... “家族“详解(从使用到底层)(1)
数据结构·算法
让学习成为一种生活方式3 小时前
DNA 甲基化综述(模式、调控与功能)--Current Opinion in Plant Biology
算法