栈和排序（字典序最大的出栈序列）

链接：https://ac.nowcoder.com/acm/problem/14893
来源：牛客网

时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 131072K，其他语言262144K
64bit IO Format: %lld

题目描述

给你一个1->n的排列和一个栈，入栈顺序给定

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

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

输入描述:

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

输出描述:

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

示例1

输入

5
2 1 5 3 4

输出

5 4 3 1 2

说明

2入栈；1入栈；5入栈；5出栈；3入栈；4入栈；4出栈；3出栈；1出栈；2出栈

要让字典序最大就要让大的数尽量先出栈。

用一个数组rmax[i]表示第 i 项到第 n 项的数的最大值。

如果栈顶元索大于第i项到第n项的最大值，那么直接让这个元索出栈，让大的先出栈总能保证字典序最大。

如果栈顶元素小于第 i 项到第 n 项的最大值，那就让该元素入栈，等着后面更大的元素。

如果最后所有元素都已经入栈了，记得还要输出栈内剩余的元素。

代码如下：

#include <bits/stdc++.h>
typedef long long LL;
#define pb push_back
#define mst(a) memset(a,0,sizeof(a))
const int INF = 0x3f3f3f3f;
const double eps = 1e-8;
const int mod = 1e9+7;
const int maxn = 1e6+10;
using namespace std;

int a[maxn];
int rmax[maxn];
stack<int> sk;
vector<int> vt;

int main()
{
    #ifdef DEBUG
    freopen("sample.txt","r",stdin); //freopen("data.out", "w", stdout);
    #endif
    
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=n;i>=1;i--)
        rmax[i] = max(a[i],rmax[i+1]);
    for(int i=1;i<=n;i++)
    {
        while(!sk.empty()&&sk.top()>=rmax[i])
        {
            vt.push_back(sk.top());
            sk.pop();
        }
        sk.push(a[i]);
    }
    while(!sk.empty())
    {
        vt.push_back(sk.top());
        sk.pop();
    }
    for(int i=0;i<vt.size();i++)
        printf(i==vt.size()-1? "%d
":"%d ",vt[i]);
    
    return 0;
}

把第一次wa的思路和代码粘下：

#include <bits/stdc++.h>
typedef long long LL;
#define pb push_back
#define mst(a) memset(a,0,sizeof(a))
const int INF = 0x3f3f3f3f;
const double eps = 1e-8;
const int mod = 1e9+7;
const int maxn = 1e5+10;
using namespace std;

stack<int> sk;
vector<int> vt;

int main()
{
    #ifdef DEBUG
    freopen("sample.txt","r",stdin); //freopen("data.out", "w", stdout);
    #endif
    
    int n;
    scanf("%d",&n);
    int MAX=n;
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        if(x!=MAX)
        {
            sk.push(x);
            continue;
        }
        vt.push_back(x);
        MAX--;
        while(!sk.empty()&&sk.top()==MAX)
        {
            vt.push_back(sk.top());
            sk.pop();
            MAX--;
        }
    }
    while(!sk.empty())
    {
        vt.push_back(sk.top());
        sk.pop();
    }
    for(int i=0;i<vt.size();i++)
        printf(i==vt.size()-1? "%d
":"%d ",vt[i]);
    
    return 0;
}

这种做法hack数据：

5

1 4 3 5 2

正确答案：5 3 4 2 1，上述输出：5 2 3 4 1

-