给定一个按照升序排列的整数数组 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]
使用两个二分查找,第一个二分找第一个出现的位置,第二个二分找最后一个出现的位置。即使使用了两个二分查找,时间复杂度仍为 log(n).
步骤:
①找第一个出现的位置,当 arr[mid] == target && target > arr[mid-1] 说明 mid 就是第一个出现的位置,否则递归调用,不断向前缩小范围,最后考虑边界,即 mid == 0,此时如果 arr[mid] == target 就返回 0,否则返回 -1.
②找最后一个出现的位置,当 arr[mid] == target && target < arr[mid + 1] 时,说明 mid 就是最后一个出现的位置,否则递归调用,不断向后缩小范围,最后考虑边界情况,即 mid == end 时,此时如果 arr[mid] == target 就返回 end,否则返回 -1.
class Solution(object):
def searchRange(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
first = self.searchFirst(nums, 0, len(nums) - 1, target)
last = self.searchLast(nums, 0, len(nums) - 1, target)
return [first, last]
def searchFirst(self, nums, start, end, target):
if end >= start:
mid = int((start + end) / 2)
if nums[mid] == target and (mid == 0 or target > nums[mid - 1]):
return mid
elif nums[mid] < target:
return self.searchFirst(nums, mid + 1, end, target)
else:
return self.searchFirst(nums, start, mid - 1, target)
return -1
def searchLast(self, nums, start, end, target):
if end >= start:
mid = int((start + end) / 2)
print(mid)
if nums[mid] == target and (mid == end or target < nums[mid + 1]):
return mid
elif nums[mid] <= target: # 注意与搜索第一个数的区别
return self.searchLast(nums, mid + 1, end, target)
else:
return self.searchLast(nums, start, mid - 1, target)
return -1
本文为互联网自动采集或经作者授权后发布,本文观点不代表立场,若侵权下架请联系我们删帖处理!文章出自:https://wangjiawei.blog.csdn.net/article/details/102786525