单调栈

P5788 【模板】单调栈 – 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

• 题意：返回数组中第一个大于第i个元素的数的下标
• 单调栈（栈中元素满足单调性）
• 从后往前遍历数组，对于当前元素，如果它比栈中的元素大，那么就不断出栈
  • 需要求当前元素对应的答案，如果栈顶元素比它还小，那么肯定不是当前的答案，也不可能是当前位置之前位置的答案，所以栈顶的这个元素就没用了，直接出栈
• 如果栈为空了，那么说明当前位置之后没有比当前位置还大的元素了，返回0。
• 如果还有，那么这个栈顶的元素肯定是比当前位置大的而且是第一个（因为在弹出栈的过程中弹出的都只是小的）
• 最后再将当前位置入栈，用于处理在这之前的元素。

// https://www.luogu.com.cn/problem/P5788
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define MAX 100000010
ll ans[MAX], datas[MAX];
stack<ll> s;
int n;
int main()
{
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        scanf("%lld", datas + i);
    }
    for (int i = n; i; i--)
    {
        while (!s.empty() && datas[i] >= datas[s.top()])
        {
            s.pop();
        }
        if (s.empty())
            ans[i] = 0;
        else
            ans[i] = s.top();
        s.push(i);
    }
    for (int i = 1; i <= n; i++)
    {
        printf("%lld ", ans[i]);
    }
}