Leetcode刷题Python35. 搜索插入位置

Posted 2022-08-02 Better Bench

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