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

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

给定一个按照升序排列的整数数组 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]

 

代码：二分法

class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        start = 0
        end = len(nums) - 1
        while start <= end:
            mid = (start+end+1)//2
            inds = mid - 1
            inde = mid + 1
            while inds >= start and nums[inds] == nums[mid]:
                inds -= 1
            while inde <= end and nums[inde] == nums[mid]:
                inde += 1
            
            if target == nums[mid]:
                return [inds+1, inde-1]
            elif target < nums[mid]:
                end = inds
            elif target > nums[mid]:
                start = inde
        return [-1, -1]