hdu-1022 Train Problem I

http://acm.hdu.edu.cn/showproblem.php?pid=1022

题意说在一个火车站只用一个铁轨可供停车，并且每次进入铁轨的车必须等后进来的车走之后才能开走，火车数量最多是9，给定一个驶入顺序，和驶出顺序，问怎么操作才能从驶入顺序得到驶出的顺序，如果能输出方案，不能就输出No。

这题跟算法竞赛入门经典讲过类似，只是这里在判断是否可行的同时需要保存一种操作状态，我是用一个数组标记，为1表示in，为0表示out，并且这题是所有车都要进入铁轨停车，所以不存在直接开走的情况。

#include<cstdio>
#include<cstring>
#include<stack>
using namespace std;
int main()
{
    int n,j,flag[10];
    char a[10],b[10];
    while(scanf("%d %s %s",&n,a+1,b+1)!=EOF)
    {
        stack<int>s;
        memset(flag,0,sizeof(flag));
        int ok=1,A=1,B=1,i=0;
        while(B<=n)
        {
            //if(a[A]==b[B]) {flag[i++]=1;A++,B++;}
            if(!s.empty()&&s.top()==(b[B]-'0')) {flag[i++]=0;s.pop();B++;}
            else if(A<=n) {flag[i++]=1;s.push(a[A]-'0');A++;}
            else {ok=0;break;}
        }
        if(ok)
        {
            printf("Yes.\n");
            for(j=0;j<i;j++)
                if(flag[j]) printf("in\n");
                else printf("out\n");
            printf("FINISH\n");
        }
        else printf("No.\nFINISH\n");
    }
    return 0;
}