列车调度(Train) 数据结构(上) Tsinghua OJ PA2

列车调度(Train)

描述

某列车调度站的铁道联接结构如Figure 1所示。

其中，A为入口，B为出口，S为中转盲端。所有铁道均为单轨单向式：列车行驶的方向只能是从A到S，再从S到B；另外，不允许超车。因为车厢可在S中驻留，所以它们从B端驶出的次序，可能与从A端驶入的次序不同。不过S的容量有限，同时驻留的车厢不得超过m节。

设某列车由编号依次为{1, 2, …, n}的n节车厢组成。调度员希望知道，按照以上交通规则，这些车厢能否以{a1, a2, …, an}的次序，重新排列后从B端驶出。如果可行，应该以怎样的次序操作?

输入

共两行。

第一行为两个整数n，m。

第二行为以空格分隔的n个整数，保证为{1, 2, …, n}的一个排列，表示待判断可行性的驶出序列{a1，a2，…，an}。

输出

若驶出序列可行，则输出操作序列，其中push表示车厢从A进入S，pop表示车厢从S进入B，每个操作占一行。

若不可行，则输出No。

Example 1

Input

5 2
1 2 3 5 4

Output

push
pop
push
pop
push
pop
push
push
pop
pop

Example 2

Input

5 5
3 1 2 4 5

Output

No

限制

1 ≤ n ≤ 1,600,000

0 ≤ m ≤ 1,600,000

时间：2 sec

空间：256 MB

算法

#include <iostream>
using namespace std;
const int MAX_NUM = 1600001;
const int MAX_NUM2 = MAX_NUM << 1;
int A[MAX_NUM], S[MAX_NUM], B[MAX_NUM];
//OP[]用于记录操作数，true为push，false为pop
//操作数个数必然是数字个数的两倍
bool OP[MAX_NUM2];
//A[]，S[]，B[]当作stack使用，OP[]当作数组使用
//当作stack使用需要设置哨兵，即数据从下标1开始储存，下标0为哨兵
int size_A, size_S, size_B, size_OP;

//下列元操作的实现使得size_X=stack_X.size()
int pop(int S[], int &size) { return S[size--]; }
int &top(int S[], int &size) { return S[size]; }
void push(int S[], int &size, const int &value)
{
    S[++size] = value;
}

int main(int argc, char const *argv[])
{
	//加速IO
    ios::sync_with_stdio(false);
    int n, m;
    cin >> n >> m;
    for (int i = n; 0 < i; --i)
        cin >> B[i];
	//哨兵A[0]为最大值n+1
    for (int i = n; 0 <= i; --i)
        A[i] = n - i + 1;
    size_A = size_B = n;
    size_S = size_OP = 0;
    //flag_fail为true即失败
    bool flag_fail = false;
    int a, b, s;
    while (size_B)
    {
        a = top(A, size_A);
        b = top(B, size_B);
        while (a <= b)
        {
            push(S, size_S, a);
            pop(A, size_A);
            if (m < size_S)
            {
                flag_fail = true;
                break;
            }
            //此时应该push
            OP[size_OP++] = true;
            a = top(A, size_A);
        }
        //若没有适宜的a可以push入S[]
        //并且S.top()!=B.top()，则失败
        if (flag_fail)
            break;
        if (size_S)
        {
            s = pop(S, size_S);
            if (s != top(B, size_B))
            {
                flag_fail = true;
                break;
            }
            //此时应该pop
            OP[size_OP++] = false;
            pop(B, size_B);
        }
    }
    if (flag_fail)
        cout << "No\n";
    else
    {
        for (int i = 0; i < size_OP; ++i)
        {
            if (OP[i])
                cout << "push\n";
            else
                cout << "pop\n";
        }
    }

    return 0;
}