题目描述:
给定一个按照升序排列的整数数组 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]:
l,r=0,len(nums)-1
res=[-1,-1]
if len(nums)==0:
return res
while(l<r):#找左边界
m=(r+l)//2
if(nums[m]>=target):
r=m
else:
l=m+1
if(nums[l]!=target):
return res
res[0]=l
r=len(nums)
while(l<r):#找右边界
m=(r+l)//2
if(nums[m]<=target):
l=m+1
else:
r=m
res[1]=l-1;
return res
p.s.也可以在找到target之后,分别从该位置向左右查找,但是特殊情况下时间复杂度为n