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

最新推荐文章于 2024-09-27 21:14:08 发布

暴躁老哥在线刷题

最新推荐文章于 2024-09-27 21:14:08 发布

阅读量2.1k

点赞数 1

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

本文链接：https://blog.csdn.net/qq_32424059/article/details/90453522

版权

Leetcode 同时被 2 个专栏收录

821 篇文章 122 订阅

订阅专栏

Python

664 篇文章 23 订阅

订阅专栏

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

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

如果数组中不存在目标值，返回 [-1, -1]。

示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]

示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]

思路：

首先确定存不存在，所以用一次二分查找来判断。

找到之后再用第二次二分查找往左边找左边界，找到的条件是已经找到下标为0了或者找到某个target，它的左边不是target。

同理进行第三次二分查找，找右边界，找到的条件是已经找到len(nums) - 1了或者找到某个target，它的右边不是target。

class Solution(object):
    def searchRange(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        lo, hi = 0, len(nums) - 1
        #找第一个target
        while(lo <= hi):
            mid = (lo + hi) // 2
            if nums[mid] == target:
                break
            elif nums[mid] > target:
                hi = mid - 1
            else:
                lo = mid + 1
        if lo > hi: #不存在
            return [-1, -1]

        midtarget = mid
        lo, hi = 0, mid
        leftpos = 0
        while(lo <= hi):
            # print lo, hi
            if (hi >= 1 and nums[hi - 1] != target) or hi == 0: #找到左边界或者找到头了
                leftpos = hi
                break
            mid = (lo + hi) // 2
            if nums[mid] == target:
                hi = mid
            elif nums[mid] < target:
                lo = mid + 1

        rightpos = 0
        lo, hi = midtarget, len(nums) - 1
        while(lo <= hi):
            if (lo <= len(nums) - 2 and nums[lo + 1] != target) or lo == len(nums) - 1: #找到右边界或者找到头了
                rightpos = lo
                break
            mid = (lo + hi + 1) // 2
            if nums[mid] == target:
                lo = mid 
            elif nums[mid] > target:
                hi = mid - 1
                
        return [leftpos, rightpos]