HDU1022 Train Problem I

第一次周赛就有这道题,现在还耿耿于怀
主要就是用到栈,水题啦

#include <cstdio>
#include <stack>
#include <cstring>

using namespace std;

const int MAX = 1000 + 10;

int n;
char in[MAX], out[MAX];
int pro[2 * MAX];                   //记录火车进出过程,1进栈0出栈

int main()
{
    while (scanf("%d", &n) == 1)
    {
        scanf("%s%s", in, out);
        stack<char> s;              //记录栈内火车
        int a = 0, b = 0;           //a记录in的个数,b记录out的个数
        int step = 0;               //记录火车一共进行多少个动作,包括入栈出栈
        int ok = 1;
        while (b < n)               //out中火车节少于n
        {
            if (in[a] == out[b])    //如果进入的火车和出去的火车节一样,直接进栈出栈
            {
                a++;
                b++;
                pro[step++] = 1;
                pro[step++] = 0;
            }
            else if (!s.empty() && s.top() == out[b])
            {
                s.pop();
                b++;
                pro[step++] = 0;
            }
            else if (a < n)
            {
                s.push(in[a++]);
                pro[step++] = 1;
            }
            else
            {
                ok = 0;
                break;
            }
        }
        printf("%s.\n", ok ? "Yes" : "No");
        if (ok)
        {
            for (int i = 0; i < step; i++)
                printf("%s\n", pro[i] ? "in" : "out");
        }
        printf("FINISH\n");
    }
    return 0;
}

从这道题学到的一点操作
stack<> s;
s.pop()出栈,s.push()入栈,s.pop()取栈顶元素,s.empty()判断栈是否为空

你可能感兴趣的:(HDU)