BZOJ3401:[USACO2009MAR]Look Up

浅谈栈:https://www.cnblogs.com/AKMer/p/10278222.html

题目传送门:https://lydsy.com/JudgeOnline/problem.php?id=3401

倒着扫,单调严格下降的栈维护最近可以看见的就行了。

时间复杂度:\(O(n)\)

空间复杂度:\(O(n)\)

代码如下:

#include <cstdio>
using namespace std;

const int maxn=1e5+5;

int n,top;
int a[maxn],stk[maxn],ans[maxn];

int read() {
	int x=0,f=1;char ch=getchar();
	for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;
	for(;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0';
	return x*f;
}

int main() {
	n=read();
	for(int i=1;i<=n;i++)
		a[i]=read();
	for(int i=n;i;i--) {
		while(top&&a[stk[top]]<=a[i])top--;
		ans[i]=stk[top];stk[++top]=i;
	}
	for(int i=1;i<=n;i++)
		printf("%d\n",ans[i]);
	return 0;
}
posted @ 2019-01-16 19:56  AKMer  阅读(103)  评论(0编辑  收藏  举报