search-insert-position

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
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值