火车进站问题-HDUOJ

火车进站问题

---

问题描述

假设杭州东火车站只有一条铁路，并且所有火车都从一侧进来，从另一侧出去。那么，如果火车A先进站，然后火车B在火车A离开之前就进站，那么火车A直到火车B离开后才能离开，可参见下图。

现在，假设车站中有n（n<=9）列火车，所有火车都有一个ID（从1到n的编号），火车以O1的顺序进站，您的任务是确定火车是否可以按O2顺序出站。

输入

输入包含几个测试用例。
每个测试用例均包含三部分：一个表示火车数量的整数和两个字符串O1和O2，其中，火车的进站顺序用O1串表示，火车的出站顺序用O2串表示。
输入在文件末尾终止，更多信息参见样例。

输出

如果不能从O1的入站顺序得到O2的出站顺序，请输出字符串“ No.”。
如果能够得到，则请输出"Yes."
然后输出进站和出站的具体方式（“in”表示火车进站，“out”表示火车出站）。
在每个测试用例之后输出一行“ FINISH”。
更多信息参见样例。

样例输入

3 123 321
3 123 312

样例输出

Yes.
in
in
in
out
out
out
FINISH
No.
FINISH

思路：火车进站先进后出问题，显然是栈的应用。题目主要的难点是关注，已经进栈的火车什么时候出栈。详细步骤，请查看下面代码中注释。

AC代码

#include <iostream>
#include <string>
#include <stack>
#include <vector>
#include <queue>

using namespace std;

int main() {
	int n = 0;
	
	// 题目已经说了以文件的结尾为标志，输入很显然以这种形式输入
	while (cin >> n) {
		string o1, o2;
		cin >> o1 >> o2;
		queue<char> q1, q2;
		stack<char> st;
        // 这里要用vector<string>将入栈和出栈的信息保存，因为不一定"Yes."
		vector<string> v;

        // 这里采用的是队列的方式，用两个队列分别装两个序列，然后先对第一个队列进行遍历
		for (int i = 0; i < n; i++) {
			q1.push(o1[i]);
			q2.push(o2[i]);
		}
        // 对第一个队列进行遍历
		while (q1.empty() == false ) {
            // q1非空时，首先要考虑的是已经进行栈中的火车，因为一旦这时候“盖住了”，后面可能就出不来了，而队列中的火车后面再进入都行。
			if (st.empty() == false && st.top() == q2.front()) {
				v.push_back("out");
				st.pop();
				q2.pop();
                // 这里的continue很重要，因为我们要保证栈顶的元素和q2队首元素不同时，才继续将q1队首的元素入栈
				continue;
			}
			char ch = q1.front();
            // 直接判断要入栈的元素(q1队首元素)和q2队首元素是否相等
			if (ch == q2.front()) {
				q1.pop();
				q2.pop();
				v.push_back("in");
				v.push_back("out");
			}
			else {
                // 不等时，先进栈
				v.push_back("in");
				st.push(ch);
				q1.pop();
			}
		}
        // q1处理完之后，现在只剩下了栈中的火车和q2了，一个一个比对即可
		while (st.empty() == false && st.top() == q2.front() ) {
			st.pop();
			q2.pop();
			v.push_back("out");
		}
        // 栈不空则比对不上，输出"No."
		if (st.empty() == false) {
			cout << "No." << endl;
		}
		else {
			cout << "Yes." << endl;
			int len = v.size();
			for (int i = 0; i < len; i++) {
				cout << v[i] << endl;
			}
		}
		cout << "FINISH" << endl;
	}
	return 0;
}