【洛谷】P5788 【模板】单调栈

题目地址：

https://www.luogu.com.cn/problem/P5788

题目描述：
给出项数为$n$的整数数列$a_{1\dots n}$。定义函数$f(i)$代表数列中第$i$个元素之后第一个大于$a_i$的元素的下标，即 f ( i ) = min ⁡ i < j ≤ n , a j > a i { j } f(i)=\min_{i<j\leq n, a_j > a_i} \{j\} f(i)=mini<j≤n,aj​>ai​​{j}。若不存在，则$f(i)=0$。试求出$f(1\dots n)$。

输入格式：
第一行一个正整数$n$。第二行$n$个正整数$a_{1\dots n}$。

输出格式：
一行$n$个整数$f(1\dots n)$的值。

数据范围：
对于$30\%$的数据，$n\le 100$；
对于$60\%$的数据，$n\le 5\times 10^3$；
对于$100\%$的数据，$1\le n\le 3\times 10^6$，$1\le a_i\le 10^9$。

开一个严格单调上升栈，如果栈非空且栈顶所指数字小于了新来的数，则找到了其右边第一个比其大的数的下标。代码如下：

#include <iostream>
using namespace std;

const int N = 3e6 + 10;
int n;
int stk[N], top;
int a[N], res[N];

int main() {
    cin >> n;
    for (int i = 1; i <= n; i++) cin >> a[i];

    for (int i = 1; i <= n; i++) {
        while(top && a[stk[top - 1]] < a[i]) {
            res[stk[top - 1]] = i;
            top--;
        }

        stk[top++] = i;
    }

    for (int i = 1; i <= n; i++) cout << res[i] << ' ';
    cout << endl;

    return 0;
}

时空复杂度$O(n)$。