LeetCode-34. 在排序数组中查找元素的第一个和最后一个位置

Kkiller233

于 2024-07-27 12:24:56 发布

阅读量203

点赞数 2

文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/Kkiller233/article/details/140733108

版权

本人算法萌新,为秋招找工作开始磨炼算法,算法题均用python实现,如果我有哪些地方做的有问题的,还请大家不吝赐教.

1.题干

给你一个按照非递减顺序排列的整数数组 nums，和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-1, -1]。

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

示例 1：
输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]
示例 2：
输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]
示例 3：
输入：nums = [], target = 0
输出：[-1,-1]
提示：

0 <= nums.length <= 105
-109 <= nums[i] <= 109
nums 是一个非递减数组
-109 <= target <= 109

2.思考

题干要求O(logn)的复杂度应该就是用二分算法了,这道题看起来是顺序排序,那么应该是二分查到的变种,问题在于二分算法是锁定一个目标值找到下标,这道题要求的是找到这个元素第一次出现的下标和最后一次出现的下标.也就是使用两次二分查找,第一次查到第一个大于等于该值的下标,第二次找到第一个大于该值的下标,然后对下标值进行比较,最后返回值

3.代码

from typing import List


class Solution:
    def binary_search(self, nums: List[int], target: int, lower: bool) -> int:
        l, r, ans = 0, len(nums) - 1, len(nums)
        while l <= r:
            mid = (l + r) // 2
            if nums[mid] > target or (lower and nums[mid] >= target):
                r = mid - 1
                ans = mid
            else:
                l = mid + 1
        return ans

    def searchRange(self, nums: List[int], target: int) -> List[int]:
        first_index = self.binary_search(nums, target, True)
        last_index = self.binary_search(nums, target, False) - 1
        if first_index <= last_index < len(nums) and nums[first_index] == target and nums[last_index] == target:
            return [first_index, last_index]
        return [-1, -1]