刷题记录:牛客NC14893栈和排序

传送门:牛客

给你一个1->n的排列和一个栈，入栈顺序给定
你要在不打乱入栈顺序的情况下，对数组进行从大到小排序
当无法完全排序时，请输出字典序最大的出栈序列
输入:
5
2 1 5 3 4
输出:
5 4 3 1 2

首先先谈一谈刚开始我对这道题所存在一些错误思路(可能有人会和我一样犯错吧):
刚开始我一看这道题目,无论能不能排序,输出的第一个肯定是最大的数字,因此我就记了一个num=n,知道遍历栈中的数字和最大值一样时才输出num,并且记录flag=当前数字在栈中的位置,然后将num–继续寻找n-1的位置,找不到则结束循环
然后分别从flag的前后开始判断那个比较大,就输出
这个思路过了一个测试点,想了一下发现有一个明显的错误之处没考虑到,就是flag后面的一个区内的最大值不一定就是在flag右边的,所以我们后来找的应该是一段区间内的最大值
所以根据这种思路,我设置了一个数组用来计算从尾部到前面的每一个区间的最大值(即后缀最大值),然后每一次都判断当前入栈的是否比后面区间的最大值还大,如果不是,则继续入栈,保证字典序最大

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
stack<int>aa;
int a[10000];
int re_max[10000];
int main() {
	int n;n=read();
	for(int i=1;i<=n;i++) {
		a[i]=read();
	}
	for(int i=n;i>=1;i--) {
		re_max[i]=max(re_max[i+1],a[i]);
	}
	for(int i=1;i<=n;i++) {
		aa.push(a[i]);
		while(aa.size()!=0&&aa.top()>re_max[i+1]) {
			cout<<aa.top()<<" ";
			aa.pop();
		}
	}
	return 0;
}