Leetcode 34
题目:
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target,返回 [-1, -1]。
进阶:你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗?
解题:
二分查找
分析:
该题目可以分解为,寻找第一个等于target的位置,和寻找第一个大于target的位置。简单的二分查找,当mid值等于target时,直接returnmid,而查找第一个位置时,当相等时,需要继续查找左区间,来检查是否还存在和target相等的值。
def searchRange(self, nums: List[int], target: int) -> List[int]:
left,right=0,len(nums)-1
# 寻找到第一个值为target的下标;
# 当mid的值等于target时,查看mid左区间,继续查找是否存在值为target的num
# 最终下标为left的num即可能为第一个位置
def findlower(left:int,right:int,nums:List[int], target: int)->int:
while left<=right:
mid = (right-left)//2 + left
num = nums[mid]
if num >=target:
right = mid-1
elif num < target:
left = mid+1
return left
# 寻找到第一个大于target的值
# 返回right,此时的right则为值等于target的最大位置。
def findhigh(left:int,right:int,nums:List[int], target: int)->int:
while left<=right:
mid = (right-left)//2 + left
num = nums[mid]
if num >target:
right = mid-1
elif num <= target:
left = mid+1
return right
low = findlower(left,right,nums,target)
high = findhigh(left,right,nums,target)
if low > high:
return [-1,-1]
else:
return [low,high]