http://www.lintcode.com/zh-cn/problem/search-for-a-range/
http://www.jiuzhang.com/solutions/search-for-a-range/
描述:
给定一个包含 n 个整数的排序数组,找出给定目标值 target 的起始和结束位置。
如果目标值不在数组中,则返回[-1, -1]
样例
给出[5, 7, 7, 8, 8, 10]和目标值target=8,
返回[3, 4]
看到题目,我们先找一下和我们做过的题目的相似之处。我们把这道题分开来看,就是要找一个给定target的第一次出现的位置和最后一次出现的位置,那么就简单了,其实就是http://www.jiuzhang.com/qa/2628/ 和 http://www.jiuzhang.com/qa/2631/ 合在一起了而已,用两次二分法求解即可。如果不理解的话,先去把之前的两道题目理解了,这道题自然就迎刃而解。
def searchRange(self, A, target):
if len(A) == 0:
return [-1, -1]
start, end = 0, len(A) - 1
while start + 1 < end:
mid = (start + end) / 2
if A[mid] < target:
start = mid
else:
end = mid
if A[start] == target:
leftBound = start
elif A[end] == target:
leftBound = end
else:
return [-1, -1]
start, end = leftBound, len(A) - 1
while start + 1 < end:
mid = (start + end) / 2
if A[mid] <= target:
start = mid
else:
end = mid
if A[end] == target:
rightBound = end
else:
rightBound = start
return [leftBound, rightBound]