算法训练-铁轨
题目描述:
某城市有一个火车站,铁轨铺设如图所示。

有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;
}
最后更新:2017-04-03 12:55:39