算法训练-铁轨

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

有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;}