栈和排序(思维)

题目：栈和排序

题目链接：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）。