leetcode 34. First and last position of element in a sorted array.

最新推荐文章于 2025-08-05 16:42:55 发布

0x7ffffff

最新推荐文章于 2025-08-05 16:42:55 发布

阅读量231

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签：二分法 leetcode python

本文链接：https://blog.youkuaiyun.com/weixin_39919480/article/details/104548507

leetcode 专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一种在O(logn)时间内查找升序数组中特定元素首次和末次出现位置的方法，通过改进的二分查找算法实现，适用于算法优化和数据搜索场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目要求

在 O(log n) 时间内查找升序数组中和给定值相等的元素的第一个和最后一个位置，找不到则返回 [-1, -1].

示例

input:  nums = [5, 7, 7, 8, 8, 10], target = 8
output: [3, 4]

按照普通二分查找方法查找，遇到 nums[i] 和 target 相等时，如果是要找最后一个位置就比较当先元素和它后一个元素的大小关系，相等则令 i = mid + 1 ，否则即为找到最右的位置；第一个位置同理。

class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:

        def leftmost_or_rightmost(is_left):

            l = len(nums)
            i = 0
            j = l - 1

            while i <= j:
                
                mid = int((i + j) / 2)

                if nums[mid] < target:
                 i = mid + 1
                elif nums[mid] > target:
                    j = mid - 1
                else:
                    
                    if not is_left: # 查找最后一个位置
                        if mid == l - 1 or nums[mid] < nums[mid + 1]:
                            return mid
                        else:
                            i = mid + 1
                            
                    else:           # 查找第一个位置
                        if mid == 0 or nums[mid] > nums[mid - 1]:
                            return mid
                        else:
                            j = mid - 1

            return -1
        return [leftmost_or_rightmost(1), leftmost_or_rightmost(0)]