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;

            

}