LeetCode刷题（35. 搜索插入位置）

最新推荐文章于 2023-06-21 23:44:23 发布

小樱桃smile

最新推荐文章于 2023-06-21 23:44:23 发布

阅读量151

点赞数

分类专栏： LeetCode 文章标签：排序数组目标值索引二分查找插入位置

本文链接：https://blog.csdn.net/qq_39172055/article/details/114435275

版权

LeetCode 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

35. 搜索插入位置

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

示例 1:
输入: [1,3,5,6], 5
输出: 2

示例 2:
输入: [1,3,5,6], 2
输出: 1

示例 3:
输入: [1,3,5,6], 7
输出: 4

示例 4:
输入: [1,3,5,6], 0
输出: 0

解析：由于给定的列表是一个排序数组，且数组中午重复元素，那么我们先可以插入目标元素，然后进行排序，找出目标元素的索引即可。

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        if target in nums:
           return nums.index(target)
        else:
            nums.append(target)
            nums.sort()
            return nums.index(target)

另外使用由于数组是有序序列，可以使用二分法进行求解：

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