给你一个按照非递减顺序排列的整数数组 nums
,和一个目标值 target
。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target
,返回 [-1, -1]
。
你必须设计并实现时间复杂度为 O(log n)
的算法解决此问题。
示例 1:
输入:nums = [5,7,7,8,8,10] , target = 8 输出:[3,4]
示例 2:
输入:nums = [5,7,7,8,8,10] , target = 6 输出:[-1,-1]
示例 3:
输入:nums = [], target = 0 输出:[-1,-1]
class Solution:
def searchRange(self, nums: List[int], target: int) -> List[int]:
memo = Counter(nums)
#去掉不存在的情况
if not target in memo:
return [-1,-1]
#找第一个位置
def findone(nums):
nums_len = len(nums)
a = 0
b = nums_len - 1
c = (a+b)//2
#当前后指针重合时,就是第一个位置
while a < b:
#当找到一个target时,不要停下继续向前查找,直到前后指针重合
if target == nums[c]:
b = c
elif target > nums[c]:
a = c + 1
else:
b = c - 1
c = (a+b)//2
return b
#找最后一个位置
def findlast(nums):
nums_len = len(nums)
a = 0
b = nums_len - 1
c = (a+b)//2
#当前后指针重合时,就是最后一个位置
while a < b:
print(a,c,b)
if target < nums[c]:
b = c - 1
#当找到一个target时,不要停下继续向后查找,直到前后指针重合
elif target == nums[c]:
a = c
else:
a = c + 1
c = (a+b+1)//2 #这里要多加一个1,保证c的更新
return a
return [findone(nums),findlast(nums)]