火车进站问题
问题描述
假设杭州东火车站只有一条铁路,并且所有火车都从一侧进来,从另一侧出去。那么,如果火车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;
}