Codility上的问题（17） PI 2012_codility 摄像头-优快云博客

本文链接：https://blog.youkuaiyun.com/caopengcs/article/details/10121573

本文介绍了一个简单的算法问题：给定一个整数数组，如何高效地计算每个元素与其右侧最近的大元素之间的距离。此问题与直方图最大矩形问题类似，通过使用单调栈实现了时间复杂度为O(N)的解决方案。

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

这个题比较简单，给定一个整数数组，对每个元素，求出和它最近比它大的数的距离（下标绝对值），如果没有比它大的数，认为距离是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;
            
}