zl程序教程

您现在的位置是:首页 >  后端

当前栏目

【Leetcode刷题Python】35. 搜索插入位置

PythonLeetCode搜索 插入 位置 刷题 35
2023-09-14 09:12:59 时间

1 题目

给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。

示例 1:

输入: nums = [1,3,5,6], target = 5
输出: 2

示例 2:

输入: nums = [1,3,5,6], target = 2
输出: 1

示例 3:

输入: nums = [1,3,5,6], target = 7
输出: 4

2 解析

用二分查找法来插入,注意边界判断,插入的位置,如果插入的target比a大,则插在a的后面。这时候拿个例子来画一下,就能懂了。
比如 nums = [1,3,5,6], target=4和target=2

3 Python实现

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