铁轨（UVa 514） 经典数据结构算法，铁轨问题

铁轨（UVa 514）

经典算法，铁轨问题
关于数据结构中栈的应用，
题目：

PopPush城市有一座著名的火车站。这个国家到处都是丘陵。而这个火车站是建于上一个世纪。不幸的是，那时的资金有限。所以只能建立起一条路面铁轨。而且，这导致这个火车站在同一个时刻只能一个轨道投入使用，因为它缺少空间，当地的惯例是每一列火车从A方向驶向B方向时候，会用某种方式将车厢重组。假设火车将要到达A方向，拥有N个车厢(N<=1000),这些车厢按照递增顺序标记为1到N。负责从组车厢的领导,必须知道是否能从组车厢让它驶出B，而这个重组的序列就是a1\a2\a3…aN.帮组他并且写一个程序来判断是否可能按照所要求的车厢顺序。你可以假设，单个的车厢可以从列车上分离出来，在他们进入站台之前。并且他们可以自由移动，知道它们上了B轨道。你也可以假设在任意时候站台可以放下无数的车厢。但是只要一个车厢进入站台，它就不能返回A轨道，同时如果它离开了站台驶向B轨道，它就不能返回站台。

所以根据题目，我们总结出一下重要的信息：
进车站的顺序是：1，2，3，4…N (N <= 1000)
出车站的循序为：a1, a2, a3, a4…an(n = =N)
根据以上的信息我们写处了以下的代码：

#include<iostream>
#include<cstdio>
#include<stack>
using namespace std;

const int MAX = 1000 + 10;

int target[MAX], n;  // target为目标车厢的顺序

int main() {
	while (scanf("%d", &n) != EOF && n >= 0) { //输入火车数目
		for (int i = 1; i <= n; i++)
			cin >> target[i];
		stack <int> s;
		int ok = 1;  //用来标记输入火车车厢的顺序是否能够正常出站
		int A = 1, B = 1;
		while (B <= n) {
			if (A == target[B]) { // 进站的火车和出站的火车为同一辆
				A++;
				B++;
			} else if (!s.empty() && target[B] == s.top()) { 
				B++; // 如果不是同一辆和栈中的车厢进行比较
				s.pop();
			} else if (A <= n) {
				s.push(A++);// A号车厢如果没有被匹配到，A号车厢进栈s
			} else {
				ok = 0; 
			// 进火车站的车厢已经没有，但是仍然不匹配目标序列，ok = 0, 自定义顺序不能正常出站
				break;
			}
		}
		cout << (ok ? "yes" : "no") << endl;
	}
}