二分查找-1--LC34.在排序数组中查找元素的第一个和最后一个位置

zeronose

于 2022-04-24 00:22:18 发布

阅读量111

点赞数

分类专栏： leetcode 文章标签： leetcode 二分查找

本文链接：https://blog.csdn.net/zeronose/article/details/124373448

版权

leetcode 专栏收录该内容

75 篇文章 0 订阅

订阅专栏

在这里插入图片描述

class Solution(object):
    def searchRange(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        # 二分查找法
        n = len(nums)
        if not n:
            return [-1, -1]
        l, r = 0, n-1
        while l<=r:
            mid = l + ((r-l) >>1)
            if nums[mid]>target:
                r = mid-1
            elif nums[mid] < target:
                l = mid + 1
            else:
                # 找到target后，还需要找到第一个target跟最后一个target
                # 先让l,r都等于mid
                # l往前找第一个等于target的，r找最后一个等于target的
                l, r = mid, mid
                while l-1>=0 and nums[l-1] == target:
                    l = l-1
                while r+1 <=n-1 and nums[r+1] == target:
                    r = r+1
                return [l,r]
        return [-1, -1]