Newcoder 58 B.栈和排序（水~）

Description

给你一个$1\to n$的排列和一个栈，入栈顺序给定

你要在不打乱入栈顺序的情况下，对数组进行从大到小排序

当无法完全排序时，请输出字典序最大的出栈序列

Input

第一行一个数$n$

第二行$n$个数，表示入栈的顺序，用空格隔开，结尾无空格

$(1\le n\le 10^6)$

Output

输出一行$n$个数表示答案，用空格隔开，结尾无空格

Sample Input

5
2 1 5 3 4

Sample Output

5 4 3 1 2

Solution

对于当前入栈元素$a_i$，如果$a_i\ge a_j,\forall j>i$，那么$a_i$是之后的最大值，故需要出栈，否则后面会有更大的元素可以增加答案的字典序，不能出栈，只需要实现维护一下后缀最大值即可，时间复杂度$O(n)$

Code

#include<cstdio>
using namespace std;
const int maxn=1000005;
int n,a[maxn],mx[maxn],ans[maxn],vis[maxn];
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	mx[n]=n;
	for(int i=n-1;i>=1;i--)
		if(a[i]<a[mx[i+1]])mx[i]=mx[i+1];
		else mx[i]=i;
	int res=0;
	for(int i=1;i<=n;i++)
		if(a[i]==a[mx[i]])ans[++res]=a[i],vis[i]=1;
	for(int i=n;i>=1;i--)
		if(!vis[i])ans[++res]=a[i];
	for(int i=1;i<=n;i++)printf("%d%c",ans[i],i==n?'\n':' ');
	return 0;
}