LeetCode刷题Python Day1 Q35：搜索插入位置-CSDN博客

本文链接：https://blog.csdn.net/JackHu12345/article/details/119831194

题目如下：

难度简单1029收藏分享切换为英文接收动态反馈

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

请必须使用时间复杂度为 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 <= 104
-104 <= nums[i] <= 104
nums 为无重复元素的升序排列数组
-104 <= target <= 104

Python风格的简洁解法：

1、利用表达式和巧妙利用统计小于的个数（来自评论区）

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        return len([i for i in nums if i < target])
        # 通过len进行选择，非常聪明！
        # 实际上将插入理解为放在所有小的元素前面，统计所有小元素的个数就行了
        # py比较方便进行比较和写这样的条件表达式
        # 但是这样不是二分查找，有损性能

2、利用index寻找，如果没有，那就插入排序再寻找

class Solution:
    def searchInsert(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        if target in nums:
            return nums.index(target)
        else:
            nums.append(target)
            nums.sort()
            return nums.index(target)
# 这个解法也很棒，python比较方便进行索引查找，如果查不到，那就先放进去调序，查找结果！

3、一个稍有难度的二分法实现（经提示修改而成）

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        high = len(nums)-1
        low = 0
        mid = low+(high-low)//2
        # 插入的位置大于左侧，小于等于右侧，采用对应的更新策略
        while (low<=high):
            if target>nums[mid]:
                low = mid+1
                mid = low+(high-low)//2
            elif target<nums[mid]:
                high = mid-1
                mid = low+(high-low)//2
            else:
                return mid
        return low

我的第一次菜鸡解答，属实麻烦，记录下来用来自嘲

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        high = len(nums)-1
        low = 0
        mid = low+(high-low)//2
        if target<=nums[0]:
            return 0
        elif target>nums[-1]:
            return high+1
        else:
        # 插入的位置大于左侧，小于等于右侧，采用对应的更新策略
            while (low<high-1):
                if target>nums[mid]:
                    low = mid
                    mid = low+(high-low)//2
                elif target<nums[mid]:
                    high = mid
                    mid = low+(high-low)//2
                else:
                    return mid
            return low+1