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

题目:
在这里插入图片描述
思路:
最开始想的比较简单,直接遍历求位置就行了,结果发现题目要求还有结束位置,普通的遍历不太好办,果断转换思路,用二分法。

二分法求这种问题就很简单了,定义区间,求区间的左右边界,最后把左右边界加入数组返回就好了。

代码如下:

class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        i = 0
        j = len(nums) - 1
        # 先确定target是否在nums中
        if target in nums:
            while i <= j:
                # 找中点
                mid = (i+j) // 2
                # 判断在左区间还是右区间,并逐步缩小
                if nums[mid] <= target:
                    i = mid + 1
                else:
                    j = mid - 1
            # 确定右边界
            right = i - 1
            if j >= 0 and nums[j] != target:
                return 0

            i = 0
            while i <= j:
                mid = (i + j) // 2
                if nums[mid] < target:
                    i = mid + 1
                else:
                    j = mid - 1
            # 确定左边界
            left = j + 1
            res = [left, right]
        else:
            res = [-1, -1]
        return res
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值