http://www.jiuzhang.com/solutions/last-position-of-target/
http://www.lintcode.com/zh-cn/problem/last-position-of-target/
描述:
给一个升序数组,找到target最后一次出现的位置,如果没出现过返回-1
样例
给出 [1, 2, 2, 4, 5, 5].
target = 2, 返回 2.
target = 5, 返回 5.
target = 6, 返回 -1.
很经典的模板题,由于要找最后一个,所以A[mid] == target的时候应该让start = mid从mid再往后找,最后要先判断end再判断start,顺序颠倒了就错了(因为要找的是最后一个)。
class Solution:
# @param {int[]} A an integer array sorted in ascending order
# @param {int} target an integer
# @return {int} an integer
def lastPosition(self, A, target):
if not A or target is None: # 先判断传给我们的数据是否为空,如果是空的话直接返回-1。如果不判断的话在后面会报错。
return -1
start = 0
end = len(A) - 1
while start + 1 < end:
mid = start + (end - start) / 2
if A[mid] < target:
start = mid
elif A[mid] > target:
end = mid
else:
start = mid # 不能直接返回,不然找的的不一定是最后一个
if A[end] == target: # 这里一定是先判断end, 因为要找的是最后一个
return end
elif A[start] == target:
return start
else:
return -1