题目:栈和排序
题目链接:https://ac.nowcoder.com/acm/problem/14893
题意:给你一个1->n的排列和一个栈,入栈顺序给定
你要在不打乱入栈顺序的情况下,对数组进行从大到小排序
当无法完全排序时,输出字典序最大的出栈序列
输入描述:
第一行一个数n
第二行n个数,表示入栈的顺序,用空格隔开,结尾无空格
输出描述:
输出一行n个数表示答案,用空格隔开,结尾无空格
样例输入:
5
2 1 5 3 4
样例输出:
5 4 3 1 2
样例解释:
2入栈;1入栈;5入栈;5出栈;3入栈;4入栈;4出栈;3出栈;1出栈;2出栈
题目分析:思维题。明显当当前元素比后面都大时,即出栈。
解题步骤:预处理数组的后缀最大值。遍历数组元素,将元素入栈,若栈顶元素大于下一个元素的后缀最大值时,则将栈顶出栈。
AC代码:
#include<iostream>
#include<stack>
using namespace std;
const int N = 1e6 + 10;
int a[N], f[N];
void solve(){
int n;
scanf("%d", &n);
for(int i = 1;i <= n;i++) scanf("%d", &a[i]);
for(int i = n;i >= 1;i--) f[i] = max(a[i], f[i - 1]);
stack<int> s;
for(int i = 1;i <= n;i++){
s.push(a[i]);
if(a[i] > f[i + 1]){
while(!s.empty() && s.top() > f[i + 1]){
printf("%d ", s.top());
s.pop();
}
}
}
}
int main(void){
solve();
return 0;
}
时间复杂度:O(n),因为每个元素最多入栈一次,出栈一次。
空间复杂度:O(n)。