LintCode - Search for a Range

最新推荐文章于 2020-03-19 15:42:00 发布

real_lisa

最新推荐文章于 2020-03-19 15:42:00 发布

阅读量155

点赞数

CC 4.0 BY-SA版权

分类专栏： LintCode

本文链接：https://blog.youkuaiyun.com/real_lisa/article/details/84227457

LintCode 专栏收录该内容

22 篇文章

订阅专栏

本文介绍了一种在有序数组中查找特定目标值首次和末次出现位置的算法，提供了两种解法，一种时间复杂度为O(nlogn)+O(n)，另一种为O(nlogn)+O(nlogn)，详细解释了每种解法的实现思路和代码。

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

解法一 O(nlogn) + O(n)

class Solution {
public:
    vector<int> searchRange(vector<int> &A, int target) {
        // lower index
        if(A.empty()) return {-1,-1};
        int n = A.size();
        int l=0, r=n-1;
        while(l<r){
            int mid = l+(r-l)/2;
            if(A[mid]<target) l=mid+1;
            else r=mid;
        }
        if(A[r]!=target) return {-1, -1};
        // upper index
        int i=r;
        while(i<n){
            if(A[++i]!=target) break;
        }
        return {r, i-1};
    }
};

解法二 O(nlogn) + O(nlogn)

结尾处如何避免check是否超出边界
Looking for lower bound(>=target): 用左闭右闭，这样结尾处只需要check nums[r]!=target
Looking for upper bound(>target): 用左闭右开，这样结尾处只需要check nums[r-1]!=target

class Solution {
public:
    vector<int> searchRange(vector<int> &A, int target) {
        // lower index
        if(A.empty()) return {-1,-1};
        int n = A.size();
        int l=0, r=n-1;
        while(l<r){
            int mid = l+(r-l)/2;
            if(A[mid]<target) l=mid+1;
            else r=mid;
        }
        if(A[r]!=target) return {-1, -1};
        // upper index
        l=r;
        int r2=n;
        while(l<r2){
            int mid = l+(r2-l)/2;
            if(A[mid]<=target) l=mid+1;
            else r2=mid;
        }
        return {r, r2-1};
    }
};