HDU 1022[Train Problem I] 栈的应用

//栈。知道什么时候应该进出站 
#include <iostream>
#include <stack>
using namespace std;
bool ans[10010];//保存进出顺序，true为"in"，false为"out" 
string in,out;

bool judge(int n){
    stack<int>st;
    int cnt=0;
    for(int i=0,j=0;i<n;i++){
        if(in[i]!=out[j]){//不是要的号码，进去 
			st.push(in[i]);
			ans[cnt++]=true;
		}
        else {
            ans[cnt++]=true,ans[cnt++]=false;//是要的号码进去直接出来 
            while(!st.empty()&&st.top()==out[++j]){//看前面有没有合适的号码 
                st.pop();
                ans[cnt++]=false;
            }
            if(st.empty())j++;//修复的BUG 
        }
    }
    if(cnt==2*n)return true;//每车都经过一进一出 
    else return false;
}

int main(){
    int n;
    while(cin>>n>>in>>out){
        if(judge(n)){
        	cout<<"Yes."<<endl;
            for(int i=0;i<2*n;i++)
            cout<<(ans[i]?"in":"out")<<endl;
        }else cout<<"No."<<endl;
        cout<<"FINISH"<<endl;
    }
    return 0;
}