python leetcode35:搜索插入位置

AII派森

于 2021-11-30 12:04:07 发布

阅读量4.9k

点赞数

分类专栏：算法文章标签： leetcode 算法职场和发展

本文链接：https://blog.youkuaiyun.com/weixin_42698464/article/details/121629558

版权

算法专栏收录该内容

27 篇文章

订阅专栏

这篇博客介绍了两种在已排序数组中查找目标值的方法。第一种是简单的线性搜索，当找到目标值时返回索引，或者根据条件判断插入位置。第二种方法使用了二分查找，通过不断缩小搜索范围来确定目标值的插入位置。示例代码展示了这两种方法的实现，并在给定的数组中查找目标值7的结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：
"""
给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。
如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
"""

法一：比较法

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        for i in range(len(nums)):
            if nums[i] == target: #相等
                return i
            elif i < len(nums)-1 and target > nums[i] and target < nums[i+1] : #target处于中间位置
                return i+1
            elif i == 0 and target < nums[i]: #开头
                return i
        return len(nums) #不存在于数组

法二：二分查找

直接套用二分法即可，即不断用二分法逼近查找第一个大于等于 \textit{target}target 的下标。

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        l,r = 0,len(nums)-1
        while l <= r:
            mid = (l+r)//2
            if target == nums[mid]:
                return mid
            elif target < nums[mid]:
                r = mid - 1
            else:
                l = mid +1
        return l


nums = [1, 3, 5, 6]
target = 7
S = Solution()
result = S.searchInsert(nums,target)
print(result)