HDU 1022 Train Problem I（栈）

Description
有n辆火车，以序列1方式进站，判断是否能以序列2方式出栈。进站不一定是一次性进入，也就是说中途可以出站
Input
多组输入，每组用例占一行包括一个整数n表示火车数量以及两个字符串表示火车进站和出战的序列，以文件尾结束输入
Output
对于每组用例，如果火车以序列1进站并可以以序列2出战则输出Yes.并输出进出站情况，否则输出No.，每组用例输出完毕后输出FINISH
Sample Input
3 123 321
3 123 312
Sample Output
Yes.
in
in
in
out
out
out
FINISH
No.
FINISH
Solution
用一个栈记录当前状态火车站中的火车序列，用k记录进栈出栈数，用ans数组记录进栈还是出栈，如果栈顶元素等于序列2中要出站的元素则将栈顶元素出栈，ans[k++]=0，否则将序列1中要进站的元素进栈，ans[k++]=1，最后通过ans数组输出进出站情况
Code

#include<cstdio>
#include<cstring>
#include<iostream>
#include<stack>
using namespace std;
int main()
{
    int n;
    char s1[1111],s2[1111];
    while(scanf("%d %s %s",&n,s1,s2)!=EOF)
    {
        stack<char>s;
        int ans[2222]={0};
        int i=0,j=0,k=1;//i,j分别为指向两个序列的指针,k为操作数 
        s.push(s1[0]);//序列1第一个元素进栈 
        ans[0]=1;//记录这次进栈操作 
        while(i<n&&j<n)
        {
            if(s.size()&&s.top()==s2[j])//栈非空且栈顶元素需要出栈 
            {
                j++;//指向序列2的指针加一 
                s.pop();//栈顶元素出栈 
                ans[k++]=0;//记录这次出栈操作 
            }
            else//不能出栈则进栈 
            {
                if(i==n)//序列1所有元素都已经进栈但并没有按序列2出栈 
                    break;
                s.push(s1[++i]);//序列1下一元素进栈 
                ans[k++]=1;//记录这次进栈操作 
            }
        }
        if(i==n)//不存在合法方案 
            printf("No.\n");
        else
        {
            printf("Yes.\n");
            for(int i=0;i<k;i++)
                if(ans[i])//ans[i]=1表示进站 
                    printf("in\n");
                else//ans[i]=0表示出站 
                    printf("out\n");
        }
        printf("FINISH\n");
    }
    return 0;
}