学习笔记 | 在排序数组中查找元素的第一个和最后一个位置

最新推荐文章于 2020-07-28 10:33:40 发布

大虎牙

最新推荐文章于 2020-07-28 10:33:40 发布

阅读量128

点赞数

分类专栏： # 数据结构 | 刷题 | 面试文章标签：二分查找

本文链接：https://blog.csdn.net/qq_34170700/article/details/107068405

版权

数据结构 | 刷题 | 面试专栏收录该内容

69 篇文章 3 订阅

订阅专栏

在这里插入图片描述

class Solution(object):
    def searchRange(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        def findlow(nums,target,low,high):
            
            while low <= high:
                mid = low + (high-low)//2
                if nums[mid] == target and (mid == 0 or nums[mid-1] < nums[mid]):
                    return mid
                if nums[mid] >= target:
                    high = mid-1
                else:
                    low = mid+1
            
            if low > high:
                return -1
            
        
        def findhigh(nums,target,low,high):
            while low <= high:
                mid = low + (high-low)//2
                if nums[mid] == target and (mid == len(nums)-1 or nums[mid+1] > nums[mid]):
                    return mid

                if nums[mid] > target:
                    high = mid-1
                else:
                    low = mid+1
            
            if low > high:
                return -1
        
        ans = []
        low = findlow(nums,target,0,len(nums)-1)
        high = findhigh(nums,target,0,len(nums)-1)
        ans.append(low)
        ans.append(high)
        return ans