leetcode-35-pythone

题目描述

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

请必须使用时间复杂度为 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
示例 4:

输入: nums = [1,3,5,6], target = 0
输出: 0
示例 5:

输入: nums = [1], target = 0
输出: 0

提示

1 <= nums.length <= 10⁴
-10⁴ <= nums[i] <= 10⁴
nums 为无重复元素的升序排列数组
-10⁴ <= target <= 10⁴

解决

没其他，第一反应就是结合二分查找（折半查找）

代码

class Solution:
    def searchInsert(self, nums: list[int], target: int) -> int:

        low, high = 0, len(nums) - 1
        while low <= high:
            mid = low + (high - low) // 2
            if nums[mid] == target:
                return mid
            elif target < nums[mid]:
                high = mid - 1
            elif target > nums[mid]:
                low = mid + 1

        nums.insert(low, target)
        return low

注意点

在获取mid索引时，使用的途径是

 mid = low + (high - low) // 2

而不是

 mid = (low + high) // 2

主要是处理溢出问题，举例，数据类型最多到10，现在两端分别是：5和7，正常相加然后整除，先由于相加12溢出报错。