题目:栈和排序

题目链接: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)。