铁轨(rails, ACM/ICPC CERC 1997,Uva 514)

题目描述

某城市有一个火车站，铁轨铺设如图所示。有n节车厢从A方向驶入车站，按进站顺序编号为1~n。你的任务是让它们按照某种特定的顺序进入B方向的铁轨并驶出车站。为了重组车厢，你可以借助中转站C。这是一个可以停放任意多节车厢的车站，但由于末端封顶，驶入C的车厢必须按照相反的顺序驶出C。对于每个车厢，一旦从A移入C，就不能再回到A了；一旦从C移入B，就不能回到C了。换句话说，在任意时刻，只有两种选择：A-->C和C-->B。

现在请你编写一个程序，对于输入指定进入B方向的铁轨并驶出车站的顺序，判断是否可行。

输入样例

5

5 4 1 2 3

5

1 2 3 4 5

6

6 5 4 3 2 1

输出样例

No

Yes

Yes

C++代码实现

#include<bits/stdc++.h>

using namespace std;

int main(){

    int n;

    while(scanf("%d",&n) == 1 && n<=1000){

        queue<int> start;

        stack<int> middle;

        int fin[n+1];

        int flag=1;

        for(int i=1;i<=n;i++){

            start.push(i);//填充初始队列

            scanf("%d",&fin[i]);

        }

        for(int i=1;i<=n;){

            int x=fin[i];

            if(middle.empty()){

                middle.push(start.front());

                start.pop();

            }

            else{//middle not empty

                if(middle.top() == x){middle.pop();i++;}

                else if(middle.top() < x){middle.push(start.front());start.pop();}

                else{flag=0;break;}

            }

        }

        printf("%s\n",flag?"yes":"no");

    }

    return 0;

}