P5788 【模板】单调栈(C++)

首先尝试的暴力解题，但是时间超限了

我们可以从前往后遍历，维护一个单调递减栈：对于每个点，如果它的值大于栈顶下标表示的数组值，就依次弹出栈顶元素、并且置这些栈顶下标对应的答案为当前点的下标。如果值小于栈顶下标对应的数组值，就压栈。最后输出答案数组即可。

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<stack>
#include<cstdlib>
#include <algorithm>
#include<string.h>
#include<math.h>
#define llu unsigned long long
using namespace std;
int main()
{
    
    
	int n,k=0,top=0;
	cin >> n ;
	long long a[n+10];
	int ans[n+5]={
    
    0},st[n+5];
	for(int i=1;i<=n;i++)
		cin >> a[i] ;
	for(int i=1;i<=n;i++)
	{
    
    
		while (top && a[st[top]] < a[i]) ans[st[top--]] = i;  
		st[++top] = i; 
	}
	for(int i=1;i<=n;i++)
		cout << ans[i] << " ";
	return 0;
}

或者从后往前遍历，维持一个单调递减栈：对于每个点 arr[i] ，弹出对应的数组值小于等于它的栈顶下标，直到栈顶下标对应的元素值 > arr[i] ，栈顶下标就是答案。然后将这个元素的下标压栈：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 3e6 + 10;
int arr[maxn], ans[maxn], st[maxn], top = -1; 

int main() {
    
    
	int n;
	scanf("%d", &n);
	for (int i = 1; i <= n; ++i) scanf("%d", &arr[i]);
	//从栈底到栈顶的下标对应的数组值(优先级)依次递减,
	//后进先出,对应->第i个元素之后第一个大于a_i的元素的下标 
	for (int i = n; i > 0; --i) {
    
    
		//将先进栈(后面的)且不大于arr[i]的值的坐标清除掉,
		//这一过程就是从i+1往后寻找第一个大于a_i的元素下标
		while (top != -1 && arr[st[top]] <= arr[i]) --top;  
		ans[i] = (top == -1 ? 0 : st[top]);	//top=-1表示之后没有大于arr[i]的值 
		st[++top] = i; //存储arr的数组下标 
	}
	for (int i = 1; i <= n; ++i) {
    
    
		if (i == 1) printf("%d", ans[i]);
		else printf(" %d", ans[i]);
	}
    return 0;
}