LeetCode - Array/Binary Search - Search Insert Position

原创于 2014-08-13 10:57:21 发布 · 363 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

30 篇文章

订阅专栏

本文深入探讨了在已排序数组中查找目标值或确定其正确插入位置的问题，通过比较两种不同时间复杂度的算法，旨在优化搜索效率并减少空间使用。详细分析了每种算法的时间复杂度、空间复杂度，并通过实例展示了它们在实际应用中的表现。

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

Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.

You may assume no duplicates in the array.

Here are few examples.
[1,3,5,6], 5 → 2
[1,3,5,6], 2 → 1
[1,3,5,6], 7 → 4

[1,3,5,6], 0 → 0

class Solution {
public:
    int searchInsert(int A[], int n, int target) {
        // 时间复杂度为O(n)，空间复杂度为O(1)
        if(A==NULL)
            return 0;
        size_t result=0;
        for(;result<n;result++)
        {
            if(A[result]>=target) return result;
        }
        return result;
    }
};

class Solution {
public:
    int searchInsert(int A[], int n, int target) {
        // 时间复杂度为O(lgn)，空间复杂度为O(1)
        if(A==NULL||n==0)
            return 0;
        int beg=0, end=n-1, result=(beg+end)/2;
        while(A[result]!=target)
        {
            if(A[result]>target)
            {
                end=result-1;
                if(result>0)result--;
            }
            if(A[result]<target)beg=++result;
            if(beg<=end)
                result=(beg+end)/2;
            else
             break;
        }
        return result;
    }
};