search-insert-position

来源:互联网 发布:乐股软件免费下载 编辑:程序博客网 时间:2024/05/10 11:39

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) # 第五种情况

0 0
原创粉丝点击