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

铁轨(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
",flag?"yes":"no");
    }
    return 0;
}