算法训练-铁轨

题目描述：
某城市有一个火车站，铁轨铺设如图所示。

有n节车厢从A方向驶入车站，按进站顺序编号1～n。
现让这些火车按照某种特定的顺序进入B方向的铁轨并驶出车站。
为了重组车厢，可以借助中转站C。
C是一个可以停放任意多节车厢的车站，但由于末端封顶，驶入C的车厢必须按照相反的顺序驶出C。
对于每个车厢，一旦从A移入C，就不能再回到A了；一旦从C移入B，就不能回到C了。
换句话说，在任意时刻，只有两种选择：A→C和C→B。
请编程判断判断：按给定的出站顺序，火车能否出站。
样例输入：
5
1 2 3 4 5
5
5 4 1 2 3
6
6 5 4 3 2 1
样例输出：
Yes
No
Yes

本题为了训练栈的应用

AC代码：

#include<stdio.h>
int a[2000],b[2000],stack[2000];
int main()
{
    int i,j,n,top,k;
    while(scanf("%d",&n)!=EOF)
    {
       for(i=0;i<n;i++)
       scanf("%d",&a[i]);//输入测试队列
       for(i=0;i<n;i++)
       b[i]=i+1;//队列初始化
       top=0;k=0;j=0;i=0;
       stack[top++]=b[0];//栈顶元素初始化
       while(k!=n)//如果出战元素刚好为输入的车厢个数，就退出
       {
          if(i==n)//只要车厢遍历完，就退出
          break;
          if(stack[top-1]!=a[j])//如果栈顶元素不等于测试元素
          {
             i++;//继续入栈
             stack[top++]=b[i];
          }
          else
          {
              k++;//如果等于出站元素，栈顶元素出栈，并且标记加1
              top--;
              j++;//测试数组向后移动
          }
       }
       if(k==n)//如果成功出战元素刚好为输入的测试车厢个数，说明出站成功
       printf("Yes\n");
       else
       printf("No\n");
    }
    return 0;
}