LeetCode_35_简单_搜索插入位置

Lins号丹

已于 2024-04-07 14:20:43 修改

阅读量20

点赞数 1

分类专栏： LeetCode进阶之路文章标签： leetcode 算法

于 2024-04-07 14:04:08 首次发布

原文链接：https://leetcode.cn/problems/search-insert-position/solutions/333632/sou-suo-cha-ru-wei-zhi-by-leetcode-solution/

版权

LeetCode进阶之路专栏收录该内容

35 篇文章 0 订阅

订阅专栏

文章目录

1. 题目
2. 思路及代码实现详解（Python）
- 2.1 二分查找

1. 题目

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

请必须使用时间复杂度为 $O (l o g n)$ 的算法。

示例 1:

输入: $n u m s = [1, 3, 5, 6], t a r g e t = 5$
输出: $2$

示例 2:

输入: $n u m s = [1, 3, 5, 6], t a r g e t = 2$
输出: $1$

示例 3:

输入: $n u m s = [1, 3, 5, 6], t a r g e t = 7$
输出: $4$

提示：

$1 <= nums.length <= 10^4$
$10^4 <= nums[i] <= 10^4$
$n u m s$ 为 无重复元素 的升序排列数组
$10^4 <= target <= 10^4$

2. 思路及代码实现详解（Python）

2.1 二分查找

考虑这个插入的位置 $p os$ ，它成立的条件为：

$n u m s [p os - 1] < t a r g e t \leq n u m s [p os]$

其中 $n u m s$ 代表排序数组。由于如果存在这个目标值，我们返回的索引也是 $p os$ ，因此我们可以将两个条件合并得出最后的目标：「在一个有序数组中找第一个大于等于 $t a r g e t$ 的下标」。如下代码，不断地更新查找的左右索引边界，只要有边界大于等于 $t a r g e t$ 就更新 $an s$ ，最后保证 $an s$ 取到的是大于等于 $t a r g e t$ 的最小值，即是该值位置。

class Solution:
    def searchInsert(self, nums, target):
        n = len(nums)
        left, right, ans = 0, n - 1, n
        while (left <= right):
            mid = ((right - left)//2) + left
            if (target <= nums[mid]):
                ans = mid
                right = mid - 1
            else:
                left = mid + 1
        return ans