Rails UVA - 514 (Stack)

PopPush城市有一座著名的火车站。这个国家到处都是丘陵。而这个火车站是建于上一个世纪。不幸的是，那时的资金有限。所以只能建立起一条路面铁轨。而且，这导致这个火车站在同一个时刻只能一个轨道投入使用，因为它缺少空间，两列火车将无路可走。具体看下图。

当地的惯例是每一列火车从A方向驶向B方向时候，会用某种方式将车厢重组。假设火车将要到达A方向，拥有N个车厢(N<=1000),这些车厢按照递增顺序标记为1到N。负责从组车厢的领导,必须知道是否能从组车厢让它驶出B，而这个重组的序列就是a1\a2\a3…aN.帮组他并且写一个程序来判断是否可能按照所要求的车厢顺序。你可以假设，单个的车厢可以从列车上分离出来，在他们进入站台之前。并且他们可以自由移动，知道它们上了B轨道。你也可以假设在任意时候站台可以放下无数的车厢。但是只要一个车厢进入站台，它就不能返回A轨道，同时如果它离开了站台驶向B轨道，它就不能返回站台。 

解题思路:

利用两个栈来模拟铁轨,首先将出栈序列填入一个栈中(注意逆序填),然后根据入栈序列将元素push到另一个栈中,每push入一个元素,便判断这个栈的栈顶元素是否与另一个栈栈顶元素相等,如果相等,则将两个栈的栈顶元素出栈,直到栈顶元素不再相等为止。

之后边push入栈序列边判断,直到所有序列push进栈中。最后,如果两个栈都为空,则输出Yes,否则输出No

凭直觉想出来的算法,我也不知道原理。。。

#include <iostream>
#include <algorithm>
#include <stack>
using namespace std;
int N;
int temparry[10010];
int main() {
	while (cin >> N) {
		if(N == 0) break;
		while (true) {
			int nums;
			bool flag = true;
			stack<int> inputStack, outPutstack;
			for (int i = 0; i < N; ++i) {
				cin >> nums;
				if (nums == 0) {
					flag = false;
					break;
				}
				temparry[i] = nums;
			}
			if (!flag) break;
			for (int s = N-1; s >= 0; --s) {
				outPutstack.push(temparry[s]);
			}
			for (int j = 1; j <= N; ++j) {
				inputStack.push(j);
				while (!inputStack.empty() && !outPutstack.empty()) {  //利用两个栈判断栈序
					if (inputStack.top() == outPutstack.top()) {
						inputStack.pop();
						outPutstack.pop();
					}
					else {
						break;
					}
				}
			}
			if (inputStack.empty() && outPutstack.empty()) cout << "Yes" << endl;
			else cout << "No" << endl;
		}
		cout << endl;
	}
	system("PAUSE");
	return 0;
}