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

一.题目描述

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
相关推荐
摘星编程7 分钟前
深入浅出 Tokio 源码:掌握 Rust 异步编程的底层逻辑
网络·算法·rust·系统编程·tokio
泡沫冰@17 分钟前
数据结构(9)
数据结构
Hard_Liquor33 分钟前
Datawhale秋训营-“大运河杯”数据开发应用创新大赛
人工智能·深度学习·算法
liu****34 分钟前
笔试强训(八)
开发语言·算法·1024程序员节
草莓工作室38 分钟前
数据结构14:查找
数据结构·算法
逐步前行1 小时前
C数据结构--线性表(顺序表|单链表|双向链表)
c语言·数据结构·链表
草莓工作室2 小时前
数据结构13:排序
c语言·数据结构·排序算法
屈冠成2 小时前
C语言数组:编辑世界的坚固桥梁
c语言·开发语言·算法
应用市场2 小时前
STM32卡尔曼滤波算法详解与实战应用
人工智能·stm32·算法
启诚科技2 小时前
树上二分(树的重心)
c++·算法·二分·树的重心