lintcode：搜索插入位置

最新推荐文章于 2022-03-21 14:58:51 发布

原创最新推荐文章于 2022-03-21 14:58:51 发布 · 280 阅读

0 ·

CC 4.0 BY-SA版权

lintcode 专栏收录该内容

28 篇文章

订阅专栏

本文介绍了一种在已排序整数数组中使用二分查找法确定目标整数插入位置的方法。该方法首先判断特殊情况，若目标小于数组最小值则直接插入数组头部；否则通过遍历数组来确定插入位置。文章提供了两种实现方式：一种为顺序查找，时间复杂度为O(n)，另一种为二分查找，时间复杂度为O(logn)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

class Solution:
    """
    @param: A: an integer sorted array
    @param: target: an integer to be inserted
    @return: An integer
    """
    def searchInsert(self, A, target):
        # write your code here
        # O(n) 1131ms
        if not A or target <= A[0]:
            return 0

        for i in range(len(A) - 1):
            if A[i] < target <= A[i+1]:
                return i + 1
        return len(A)

class Solution:
    """
    @param: A: an integer sorted array
    @param: target: an integer to be inserted
    @return: An integer
    """
    def searchInsert(self, A, target):
        # write your code here
        # 二分查找 O(logn) 556ms
        if not A or target <= A[0]:
            return 0
        n = len(A)
        # 其余情况，反向遍历列表，找到第一个小于等于target的数
        for i in range(-1,-n,-1):
            if A[i] > target: # 大于target,继续循环
                continue
            elif A[i] == target: # 等于，返回n+i;因为i是负数,反向循环
                return n+i
            else:
                return n+i+1 # 小于target，在n+i位置之后插入元素，返回n+i+1