LintCode : 插入搜索位置

最新推荐文章于 2019-05-22 12:15:19 发布

orekinana

最新推荐文章于 2019-05-22 12:15:19 发布

阅读量267

点赞数

CC 4.0 BY-SA版权

分类专栏： LintCode 文章标签：搜索

本文链接：https://blog.youkuaiyun.com/orekinana/article/details/59676315

LintCode 专栏收录该内容

34 篇文章

订阅专栏

本文介绍了一个算法问题：如何在一个已排序的数组中查找特定目标值的插入位置。使用二分查找法，文章详细解释了如何高效地确定目标值应该插入的位置，并提供了具体的代码实现。

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

LintCode : 插入搜索位置

题目

给定一个排序数组和一个目标值，如果在数组中找到目标值则返回索引。如果没有，返回到它将会被按顺序插入的位置。你可以假设在数组中无重复元素。

样例

[1,3,5,6]，5 → 2

[1,3,5,6]，2 → 1

[1,3,5,6]， 7 → 4

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

思路

通过二分法，我们找到一个比目标值小的最大元素，得到他的下标，返回下标+1。如果找到这个元素将直接返回下标。

代码

int searchInsert(vector<int> &A, int target) {
    int p = 0;
    int r = A.size() - 1;
    if(A.size() == 0 || target < A[0]) {
        return 0;
    }
    while(p <= r) {
        if(A[(p + r) / 2] == target) {
            return (p + r) / 2;
        } 
        else if(A[(p + r) / 2] > target) {
            r = (p + r) / 2 - 1;
        }
        else {
            p = (p + r) / 2 + 1;
        }
    }
    if(A[p] > target) {
        return p;
    }
    return p + 1;
}