二分查找算法——搜索插入位置

一.题目描述

35. 搜索插入位置 - 力扣(LeetCode)

二.题目解析

如果只看题目的前半句,直接就用简单二分秒了。但是题目还有后半句,如果找不到目标值,则返回它应该插入的位置。

因为数组是排序的,所以应该插入的位置就是target插入之后,依旧满足数组是有序的。

三.算法原理

1.暴力解法

我们先遍历一边数组,看target是否存在。如果不存在,我们就在遍历一边数组,找到第一个大于target的元素,返回该元素的下标即可。

时间复杂度:遍历了两边数组,所以时间复杂度为O(N)

空间复杂度:遍历过程中只是用了有限个变量,所以空间复杂度为O(1)

2.二分算法

查找一个数在不在我们可以用简单二分来解决,但是这道题涉及到如果不存在,需要返回待插入的位置,所以简单二分是解决不了的。

我们可以对数组进行分析,target会将数组分为两个部分,[0~target]<=target,[target+1,size-1]>target。这样的划分规则就类似于二分查找中------查找区间右端点,所以我们可以根据此二段性,利用查找区间的右端点来解决问题。

查找完成后,会剩余一个元素,该元素要么是target,要么不是。如果不是的话,又分为两种情况,当前元素小于target,此时我们就应该插入下一个位置;如果当前位置>target,说明插入到该元素位置即可。

四.代码实现

复制代码
// C++

class Solution 
{
public:
    int searchInsert(vector<int>& nums, int target) 
    {
        int left = 0;
        int right = nums.size()-1;
        while(left<right)
        {
            int mid = left+(right-left+1)/2;
            if(nums[mid] > target)
            {
                right = mid-1;
            }
            else
            {
                left = mid;
            }
        }
        if(nums[left] < target)
        {
            return left+1;
        }
        return left; 
    }
};

# python

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        left,right = 0,len(nums)-1
        while left<right:
            mid = left+(right-left+1)//2
            if nums[mid] > target:
                right = mid-1
            else:
                left = mid
        if(nums[left] < target):
            return left+1
        return left
相关推荐
YuTaoShao8 分钟前
【LeetCode 热题 100】994. 腐烂的橘子——BFS
java·linux·算法·leetcode·宽度优先
Wendy14418 小时前
【线性回归(最小二乘法MSE)】——机器学习
算法·机器学习·线性回归
拾光拾趣录8 小时前
括号生成算法
前端·算法
渣呵9 小时前
求不重叠区间总和最大值
算法
拾光拾趣录9 小时前
链表合并:双指针与递归
前端·javascript·算法
好易学·数据结构9 小时前
可视化图解算法56:岛屿数量
数据结构·算法·leetcode·力扣·回溯·牛客网
香蕉可乐荷包蛋10 小时前
AI算法之图像识别与分类
人工智能·学习·算法
chuxinweihui10 小时前
stack,queue,priority_queue的模拟实现及常用接口
算法
tomato0910 小时前
河南萌新联赛2025第(一)场:河南工业大学(补题)
c++·算法
墨染点香11 小时前
LeetCode Hot100【5. 最长回文子串】
算法·leetcode·职场和发展