算法入门经典第六章 例题6-2 铁轨

题目描述

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

输入

输入可能有多组数据，每组数据的第一行为一个整数N（N<=1000)，第二有N个整数，分别为指定的进入B方向铁轨并驶出车站的顺序。

输出

对于每组数据，输出仅有一行，如果可以按指定的顺序驶出车站，则输出Yes，否则输出No。

样例输入

5
5 4 1 2 3
5
1 2 3 4 5 
6
6 5 4 3 2 1

样例输出

No
Yes
Yes

#include<cstdio>
#include<stack>
using namespace std;
const int maxn=1000+10;

int n,target[maxn];
int main()
{
    while(scanf("%d",&n)==1)
    {
        stack<int> s;
        int A=1,B=1;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&target[i]);
        }
        int ok=1;
        while(B<=n)
        {
            if(A==target[B]) {A++;B++;}
            else if(!s.empty()&&s.top()==target[B]) {
                s.pop(); B++;
            }
            else if(A<=n) s.push(A++);
            else{
                ok=0;break;
            }
            }
            printf("%s
",ok?"Yes":"No");
        }
         
    return 0;
}