hdu--1022--Train Problem I

题目链接

#include <iostream> 
#include <stack>
using namespace std;
int main() {
    int n;
    while (cin>>n) {
        string a, b;
        cin>>a>>b;
        string tag;
        stack<int> s;
        for (int i=0, j=0; i<a.size(); i++) {
        /*
            如果a[i]  = b[j]说明是进去再出来，如果不相等，说明进去没出来
            如果遇到相等的，那么判断栈空，不空则判断后面出站的是否是栈顶元素，如果是继续出栈
            123 321
            匹配到3，判断2=栈顶2，1=栈顶1，判断完毕，YES 
        */ 
            if (a[i] != b[j]) {
                tag += "0";
                s.push(a[i]);
            } else {
                tag += "01";
                j++;
                while (!s.empty()) {
                    if (s.top() == b[j]) {
                        tag += "1";
                        s.pop();
                        j++;
                    } else break;
                }
            }
        }
        if (s.empty()) {
            cout<<"Yes."<<endl;
            for (int i=0; i<tag.size(); i++) {
                if (tag[i] == '1') {
                    cout<<"out
";
                } else {
                    cout<<"in
";
                }
            }
            cout<<"FINISH
";
        } else {
            cout<<"No.
FINISH
";
        }
    }
    return 0;
}