剑指offer-53.在排序数组中查找数字

题目来源:(两道题仅返回值不同)

https://leetcode-cn.com/problems/zai-pai-xu-shu-zu-zhong-cha-zhao-shu-zi-lcof/

https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/

给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。

你的算法时间复杂度必须是 O(log n) 级别。 

思路:

看到“排列数组”,第一反应就得是二分查找。但是一般的二分,如果在中点处刚好找到一个数等于target,那接下来往哪边走呢?没办法判断,所以考虑用两次二分,一次找起始下标,一次找结束下标。(注意找结束下标时mid=(low+high+1)//2,以及一些特殊情况的判断)

class Solution(object):
    def searchRange(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        low, high = 0, len(nums)-1         # 第一次二分,找起始下标
        while low < high:
            mid = (low + high) // 2
            if nums[mid] >= target:
                high = mid
            else:
                low = mid + 1
        
        if not nums or nums[low] != target:
            return [-1, -1]
        
        left = low
        low, high = 0, len(nums)-1         # 第二次二分,找结束下标
        while low < high:
            mid = (low + high + 1) // 2      # 注意+1,不然会死循环
            if nums[mid] <= target:
                low = mid
            else:
                high = mid - 1
        right = high
        return [left, right]        # 统计次数时返回right-left+1

参考:

https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/solution/python3-liang-ci-er-fen-cha-zhao-jian-shao-ni-de-s/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值