LintCode : 搜索区间

最新推荐文章于 2019-08-23 00:03:16 发布

原创最新推荐文章于 2019-08-23 00:03:16 发布 · 341 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#lintcode

lintcode 专栏收录该内容

139 篇文章

订阅专栏

本文介绍了一种在有序数组中寻找特定元素起始与结束位置的算法。提供了两种解决方案：一种使用双指针法，时间复杂度为O(n)；另一种采用二分查找法，效率更高。适用于目标值存在的情况，若不存在则返回[-1, -1]。

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

LintCode : 搜索区间

给定一个包含 n 个整数的排序数组，找出给定目标值 target 的起始和结束位置。

如果目标值不在数组中，则返回[-1, -1]

方案一：

指定一高一低两个指针，低指针向后遍历，高指针向前遍历，最终低指针指向起始位置，高指针指向结束位置。时间复杂度O(n)。


class Solution {
    /** 
     *@param A : an integer sorted array
     *@param target :  an integer to be inserted
     *return : a list of length 2, [index1, index2]
     */
public:
    vector<int> searchRange(vector<int> &A, int target) {
        // write your code here
        int low = 0;
        int high = A.size() - 1;
        vector<int> ans;
        while(low < A.size()){
            while(low < A.size() && A[low] != target) low ++;
            while(0 < high && A[high] != target) high --;
            if (low < A.size()){
                ans.push_back(low);
                ans.push_back(high);
                return ans;
            }
        }
        ans.push_back(-1);
        ans.push_back(-1);
        return ans;

    }
};

方案二：

二分查找。


class Solution {
    /** 
     *@param A : an integer sorted array
     *@param target :  an integer to be inserted
     *return : a list of length 2, [index1, index2]
     */
public:
    vector<int> searchRange(vector<int> &A, int target) {
        // write your code here
        int low = lower_bound(A.begin(), A.end(), target) - A.begin();
        int high = upper_bound(A.begin(), A.end(), target) - A.begin();
        vector<int> ans;
        if(low == high){
            low = -1;
            high = 0;
        }
        ans.push_back(low);
        ans.push_back(high - 1);

        return ans;

    }
};