力扣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 的位置,也就是最终的插入位置
相关推荐
圣保罗的大教堂11 小时前
leetcode 1752. 检查数组是否经排序和轮转得到 简单
leetcode
weixin_4684668511 小时前
PyTorch 与 TensorFlow 实战选型与应用场景指南
人工智能·pytorch·深度学习·算法·机器学习·tensorflow·深度学习框架
x_xbx11 小时前
LeetCode:647. 回文子串
算法·leetcode·职场和发展
Dlrb121111 小时前
数据结构-树与二叉树
数据结构·二叉树·深度优先··广度优先·层序遍历
Chen_harmony12 小时前
二十二、动态内存管理
c语言·数据结构·算法
Black蜡笔小新12 小时前
制造业AI质检工作站/自动化AI算法训练服务器DLTM企业AI算力工作站筑牢制造业品质防线
人工智能·算法·自动化
晚风予卿云月12 小时前
【模拟】多项式输出 & 蛇形方阵 & 字符串展开
c++·算法·模拟算法·随笔·竞赛练习
listhi52012 小时前
基于MATLAB的自适应粒子群算法(APSO)实现大规模分类特征选择
算法·matlab·分类
weixin_4074438712 小时前
基于Sentinel-1/2数据特征优选的冬小麦识别
人工智能·算法·随机森林·机器学习·sentinel