Rails UVA - 514 (Stack)

客官°小女子只卖身不卖艺 2022-04-23 13:34 257阅读 0赞

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

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

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2FsZXgxOTk3MjIy_size_16_color_FFFFFF_t_70][]

解题思路:

利用两个栈来模拟铁轨,首先将出栈序列填入一个栈中(注意逆序填),然后根据入栈序列将元素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;
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2FsZXgxOTk3MjIy_size_16_color_FFFFFF_t_70]: /images/20220310/6322ee0c50de4f9e97459980f7fcbc17.png