35. 搜索插入位置

问题链接

https://leetcode.cn/problems/search-insert-position/description/

解题思路

搜索插入位置,是一个常见的二分算法。

二分是有固定模板的。这个题目是搜索插入位置,和最简单的二分不同。

不同之处在于,它在查找失败时不返回False,而是返回插入位置,这就要求我们能够优雅的查找到插入位置。

首先我们按照二分的一般思路写好代码。然后我们去想,如果此时left == right ,即这是最后一轮比较的话,如果target < mid,那代表target应该插入mid当前的位置上,此时指向mid的是left和mid。

如果target > mid , 此时target应该插入在mid后,此时指向mid+1的只有left。所以我们直接返回left即可。

代码

复制代码
class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        left = 0
        right = len(nums)-1
        while left <= right:
            mid = (left + right) >> 1
            if nums[mid] == target:
                return mid
            elif nums[mid] < target:
                left = mid + 1
            else:
                right = mid - 1
        return left
复制代码

 

posted @   BJFU-VTH  阅读(29)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
点击右上角即可分享
微信分享提示