单调栈（洛谷）

最新推荐文章于 2024-03-06 16:45:19 发布

scj2021

最新推荐文章于 2024-03-06 16:45:19 发布

阅读量451

点赞数 3

文章标签： c语言

本文链接：https://blog.youkuaiyun.com/scj2021/article/details/120606929

版权

本文介绍了如何使用单调栈解决洛谷P5788题目，思路是维护一个单调递减的栈，遇到比栈顶元素小的数就入栈，并记录对应函数值，最后弹出栈顶元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

P5788（洛谷）
P5788 单调栈
思路：
维护一个单调递减的栈，重复执行：当入栈的下标所对应的值，那么入栈的下标为栈顶下标的f（x）的值，存入ans函数，并弹出栈顶元素。
代码：

#include<bits/stdc++.h>
using namespace std;
stack<int> w;
int a[3000005],ans[3000005];
int n;
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(int i=1;i<=n;i++){
		while(!w.empty()){
			if(a[w.top()]<a[i]){
				ans[w.top()]=i;
				w.pop();
			}
			else break;
		}
		w.push(i);
	}
	while(!w.empty()){
		ans[w.top()]=0;
		w.pop();
	}
	for(int i=1;i<=n;i++)
		cout<<ans[i]<<" ";
	return 0;
}