python--hot 100--简单--1--34. 在排序数组中查找元素的第一个和最后一个位置

在这里插入图片描述
方法:核心思想是使用二分查找

代码:

from typing import List


class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        l, r, mid = 0, len(nums) - 1, 0

        # 查找一个target在nums中的位置
        while l <= r:
            mid = (l + r) >> 1
            if target < nums[mid]:
                r = mid - 1
            elif target > nums[mid]:
                l = mid + 1
            else:
                break

        if l > r:
            return [-1, -1]

        # 在(l, mid]中找左边界,该区域内的元素均小于等于nums[mid]
        r_ = mid - 1
        while l <= r_:
            mid_ = (l + r_) >> 1
            if target == nums[mid_]:
                r_ = mid_ - 1
            else:  # target > nums[mid_]的情况,不可能有target < nums[mid_]
                l = mid_ + 1

        # 在[l_, mid)中找右边界,该区域内的元素均大于等于nums[mid]
        l_ = mid + 1
        while l_ <= r:
            mid_ = (l_ + r) >> 1
            if target == nums[mid_]:
                l_ = mid_ + 1
            else:  # target < nums[mid_]的情况,不可能有target > nums[mid_]
                r = mid_ - 1
        return [l, r]


if __name__ == '__main__':
    solution = Solution()
    nums = [5, 7, 7, 8, 8, 10]
    target = 7
    nums = [5, 7, 7, 8, 8, 10]
    target = 6
    res = solution.searchRange(nums, target)
    print(res)


在这里插入图片描述

[题目源于leetcode hot 100]

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值