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

 

鏈接：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\n":"%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\n":"%d ",vt[i]);
    
    return 0;
}

這種做法hack數據：

5

1 4 3 5 2

正確答案：5 3 4 2 1，上述輸出：5 2 3 4 1

 

 

 

 

-