一、题设
给你一个按照非递减顺序排列的整数数组 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]
二、基本思路
随便找一个target,然后前后分别往左右边探索就可以。
三、代码实现
def searchRange(self, nums: List[int], target: int) -> List[int]:
left,right,flag= 0, len(nums)-1,1
if len(nums) == 0:
return [-1,-1]
if nums[0] == target and len(nums) == 1:
return [0,0]
# 折半随便找一个target
while left <= right and flag:
mid_index = int ((left + right) / 2)
if nums[mid_index] == target:
flag = 0
elif nums[mid_index] < target:
left = mid_index + 1
else:
right = mid_index - 1
if flag:
return [-1,-1]
j1 = mid_index
j2 = mid_index
# 往前找第一个target
while j1 >= 0:
if nums[j1] != target:
break
j1 -= 1
# 往后找最后一个target
while j2 < len(nums):
if nums[j2] != target:
break
j2 += 1
return ([j1+1,j2-1] if j1+1 < j2-1 else [j2-1,j1+1])