http://www.lintcode.com/zh-cn/problem/search-insert-position/
http://www.jiuzhang.com/solutions/search-insert-position/
描述:
给定一个排序数组和一个目标值,如果在数组中找到目标值则返回索引。如果没有,返回到它将会被按顺序插入的位置。
你可以假设在数组中无重复元素。
样例
[1,3,5,6],5 → 2
[1,3,5,6],2 → 1
[1,3,5,6], 7 → 4
[1,3,5,6],0 → 0
其实吧,这道题就是C++标准库里面的lower_bound的实现。。。
思路是用二分法去不断逼近在A中接近target的值,最后有五种情况:
1、A[start] == target
2、A[end] == target
3、target < A[start] < A[end] 这个情况就是数组中所有的数都比target大,target要插入到第一个
4、A[start] < target < A[end]
5、A[start] < A[end] < target 这个情况就是数组中所有的数都比target小,target要插入到end后面,就是最后一个
套用二分法模板,最后对这五种情况进行判断即可。
class Solution:
"""
@param A : a list of integers
@param target : an integer to be inserted
@return : an integer
"""
def searchInsert(self, A, target):
if len(A) == 0: # 照例先判断特殊情况,不判断后面会报错,下标越界
return 0
start, end = 0, len(A) - 1
# first position >= target
while start + 1 < end:
mid = (start + end) / 2
if A[mid] >= target:
end = mid
else:
start = mid
if A[start] >= target: # 第一种可以和第三种合并在一起
return start
if A[end] >= target: # 第二种可以和第四种合并在一起,因为如果走到这步了就说明target > A[start]
return end
return len(A) # 第五种情况