算法题：单调栈

问题描述

给出项数为 $n$ 的整数数列 $a_{1 \dots n}$。

定义函数 $f(i)$ 代表数列中第 $i$ 个元素之后第一个大于 $a_i$ 的元素的下标，即 $f(i)=\min_{i<j\leq n, a_j > a_i} \{j\}$。若不存在，则 $f(i)=0$。

试求出 $f(1\dots n)$。

输入格式

第一行一个正整数 $n$。

第二行 $n$ 个正整数 $a_{1\dots n}$。

输出格式

一行 $n$ 个整数 $f(1\dots n)$ 的值。

样例

输入

5
1 4 2 3 5

输出

2 5 4 5 0

数据范围

对于 $30\%$ 的数据，$n\leq 100$；

对于 $60\%$ 的数据，$n\leq 5 \times 10^3$ ；

对于 $100\%$ 的数据，$1 \le n\leq 3\times 10^6$，$1\leq a_i\leq 10^9$。

解决方案

思路

单调栈

代码

#include <iostream>
#include <stack>

using namespace std;

const int N = 3e6 + 5;

int n;
int arr[N];
int result[N];

int main() {

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

    stack<int> s;
    for (int i = n; i >= 1; --i) {
        while (!s.empty() && arr[i] >= arr[s.top()])
            s.pop();

        if (s.empty())
            result[i] = 0;
        else
            result[i] = s.top();

        s.push(i);
    }

    for (int i = 1; i <= n; ++i) {
        cout << result[i] << " ";
    }

    return 0;
}