算法入门经典第六章 例题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\n",ok?"Yes":"No"); } return 0; }