描述
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值,返回 [-1, -1]。
在线评测地址:[领扣题库官网](https://www.lintcode.com/problem/1536/?utm_source=sc-tianchi
-sz-0514)
样例1
输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]
样例2
输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]
解题思路
不用烧脑, 直接两次无脑二分,求left 和 right。
不过需要注意的是
if numsmid <= target: start = mid else: end = mid 得到是right, 我第一次以为是left。 感觉如果再碰到,还是有可能写这个bug出来,究竟怎么想会好一点呢?大家有什么方法?
源代码
class Solution:
def searchRange(self, nums, target):
if not nums:
return [-1, -1]
left, right = -1, -1
start, end = 0, len(nums) - 1
while start + 1 < end:
mid = (start + end) // 2
if nums[mid] <= target:
start = mid
else:
end = mid
if nums[start] == target:
right = start
if nums[end] == target:
right = end
start, end = 0, len(nums) - 1
while start + 1 < end:
mid = (start + end) // 2
if nums[mid] >= target:
end = mid
else:
start = mid
if nums[end] == target:
left = end
if nums[start] == target:
left = start
if left == -1 and right == -1:
return [-1, -1]
return [left, right]
更多题解参考:九章官网solution