Codility上的问题 (17) PI 2012

这个题比较简单,给定一个整数数组,对每个元素,求出和它最近比它大的数的距离(下标绝对值),如果没有比它大的数,认为距离是0。

数组元素个数 N [0..50000],数组元素范围[-10^9, +10^9]。

要求复杂度 时间 空间 都是O(N)。

分析:

    这个题比较简单吧。跟直方图最大矩形差不多,类似于求左右边界。求左边界的话记住,有这个数在的话,比它更早的并且比它小的数都没有意义(因为有这个数存在,而它又很大,右边的数往左找的话会先选择这个数)。于是栈内元素是单调递减的,求右边界类似。 


代码:

// you can also use includes, for example:
// #include <algorithm>

vector<int> solution(const vector<int> &A) {
    // write your code here...
    int n = A.size(),i;
    vector<int> a,r;
    r.resize(n);
    for (i = 0; i < n; ++i) {
        while ((!a.empty()) && (A[a.back()] <= A[i])) {
            a.pop_back();
        }
        r[i] = a.empty()?n:(i - a.back());
        a.push_back(i);
    }
    a.clear();
    for (i = n - 1; i >= 0; --i) {
        while ((!a.empty()) && (A[a.back()] <= A[i])) {
            a.pop_back();
        }
        if (!a.empty()) {
            r[i] = min(r[i], a.back() - i);
        }
        else if (r[i] == n) {
            r[i] = 0;
        }
        a.push_back(i);
    }
    return r;
            
}

 

 

 

posted on 2013-08-20 20:31  you Richer  阅读(175)  评论(0编辑  收藏  举报