LeetCode_34_中等_在排序数组中查找元素的第一个和最后一个位置

Lins号丹

已于 2024-04-07 14:19:59 修改

阅读量361

点赞数 5

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

于 2024-04-07 11:40:52 首次发布

原文链接：https://leetcode.cn/problems/find-first-and-last-position-of-element-in-sorted-array/description/

版权

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

35 篇文章 0 订阅

订阅专栏

文章目录

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

1. 题目

给你一个按照非递减顺序排列的整数数组 $n u m s$ ，和一个目标值 $t a r g e t$ 。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 $t a r g e t$ ，返回 $[- 1, - 1]$ 。

你必须设计并实现时间复杂度为 $O (l o g n)$ 的算法解决此问题。

示例 1：

输入： $n u m s = [5, 7, 7, 8, 8, 10], t a r g e t = 8$
输出： $[3, 4]$

示例 2：

输入： $n u m s = [5, 7, 7, 8, 8, 10], t a r g e t = 6$
输出： $[- 1, - 1]$

示例 3：

输入： $n u m s = [], t a r g e t = 0$
输出： $[- 1, - 1]$

提示：

$0 <= nums.length <= 10^5$
$10^9 <= nums[i] <= 10^9$
$n u m s$ 是一个非递减数组
$10^9 <= target <= 10^9$

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

2.1 二分查找

在排序数组中找数，以及看到复杂度要求为 $O (l o g n)$ ，则直接考虑用二分查找的方法，该方法在具体问题中的实现有所差异。具体到该题目，考虑 $t a r g e t$ 开始和结束位置，其实我们要找的就是数组中「第一个等于 $t a r g e t$ 的位置」（记为 $l e f t I d x$ ）和「第一个大于 $t a r g e t$ 的位置减一」（记为 $r i g h t I d x$ ）。

二分查找中，寻找 $l e f t I d x$ 即为在数组中寻找第一个大于等于 $t a r g e t$ 的下标，寻找 $r i g h t I d x$ 即为在数组中寻找第一个大于 $t a r g e t$ 的下标，然后将下标减一。两者的判断条件不同，为了代码的复用，我们定义 binarySearch(nums, target, lower) 表示在 $n u m s$ 数组中二分查找 $t a r g e t$ 的位置，如果 $l o w er$ 为 $t r u e$ ，则查找第一个大于等于 $t a r g e t$ 的下标，否则查找第一个大于 $t a r g e t$ 的下标。

最后，因为 $t a r g e t$ 可能不存在数组中，因此我们需要重新校验我们得到的两个下标 $l e f t I d x$ 和 $r i g h t I d x$ ，看是否符合条件，如果符合条件就返回 $[l e f t I d x, r i g h t I d x]$ ，不符合就返回 $[- 1, - 1]$ 。

class Solution:
    def searchRange(self, nums, target):

        def binarySearch(nums, target, lower):
            left = 0
            right = len(nums) - 1
            ans = len(nums)
            while left <= right:
                mid = (left + right) // 2
                if (nums[mid] > target) or (lower and nums[mid] >= target):
                    right = mid - 1
                    ans = mid
                else:
                    left = mid + 1
            return ans 

        left_b = binarySearch(nums, target, True)
        right_b = binarySearch(nums, target, False) - 1
        if (left_b <= right_b) and (right_b < len(nums)) and (nums[left_b] == target) and (nums[right_b] == target):
            return [left_b, right_b]
        else:
            return [-1, -1]