P5788 【模板】单调栈

5788
题目背景
模板题，无背景。

2019.12.12 更新数据，放宽时限，现在不再卡常了。

题目描述
给出项数为 nn 的整数数列 a_{1 \dots n}a
1…n
​
。

定义函数 f(i)f(i) 代表数列中第 ii 个元素之后第一个大于 ai
的元素的下标，即 f(i)=\min_{i<j\leq n, a_j > a_i} {j}f(i)=min
i<j≤n,a
j
​

a
i
​

​
{j}。若不存在，则 f(i)=0f(i)=0。

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

输入格式
第一行一个正整数 nn。

第二行 nn 个正整数 a_{1\dots n}a
1…n
​
。

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

输入输出样例
输入 #1复制
5
1 4 2 3 5
输出 #1复制
2 5 4 5 0

#include <iostream>
#include <stack>
#include <cstdio>
using namespace std;
const int maxn = 3e6 + 5;
int a[maxn], res[maxn];
int s[maxn];
int main()
{
   int n;
   scanf("%d", &n);
   for (int i = 1; i <= n; i++)
   {
      scanf("%d", &a[i]);
   }
   int top = 0;
   for (int i = n; i >= 1; i--)
   {
      while (top && a[s[top]] <= a[i])
         top--;
      if (!top)
         res[i] = 0;
      else
         res[i] = s[top];
      s[++top] = i;
   }
   for (int i = 1; i <= n; i++)
   {
      printf("%d ", res[i]);
   }
   return 0;
}