题意:某城市有一个火车站,铁轨铺设如图所示。有n节车厢从A方向驶入车站,按建站顺序编号为1~n。你的任务是判断是否能让他们按照某种特定的顺序进入B方向的铁轨并驶出车站。例如,出战顺序(5 4 1 2 3)是不可能的,但(5 4 3 2 1)是可能的。
为了重组车厢,你可以借助中转站C。这是一个可以停放任意多节车厢的车站,但由于末端封顶,驶入C的车厢必须按照相反的顺序驶出C。对于每个车厢,一旦从A一如C,就不能再回到A了;一旦从C移入B,就不能回到C了。换句话说,在任意时刻,只有两种选择:A->C和C->B。
分析:在中转站C中,车厢符合后进先出的原则,因此是一个栈。
以上内容来自算法竞赛入门经典。
#include<bits/stdc++.h> using namespace std; const int MAXN = 1005; int n, t[MAXN]; int main() { while (~scanf("%d", &n) && n) { while (~scanf("%d", &t[1]) && t[1]) { stack<int> S; int A = 1, B = 1; for (int i = 2; i <= n; i++) scanf("%d", &t[i]); bool ok = true; while (B <= n) { if (A == t[B]) { A++; B++; } else if (!S.empty() && S.top() == t[B]) { S.pop(); B++; } else if(A <= n) S.push(A++); else { ok = false; break; } } printf("%s\n", ok ? "Yes" : "No"); } printf("\n"); } return 0; }