HDU1022 Train Problem I 栈的应用

原题链接在这里

这道题主要用到了 STL 模板中的栈。栈的常用操作如下：

#include <stack>  //头文件

stack <char> s; //建立一个元素为 char 类型的栈 
s.empty()       //如果栈为空返回 true，否则返回 false  
s.size()        //返回栈中元素的个数  
s.pop()         //删除栈顶元素但不返回其值  
s.top()         //返回栈顶的元素，但不删除该元素  
s.push(X)       //在栈顶压入新元素 ，参数 X 为要压入的元素

在这道题中，由于需要实现随时进出栈，所以可以每从进站顺序 O1 压入一个新的元素，先让标记数组 rcd[] 打上进站标记，然后比较当前栈顶元素是否和要求出站顺序 O2 中的元素相同，若是相同，则 s.pop() 、 O2 的下标 +1 、 标记数组 rcd[] 打上出站标记。

代码如下：

#include <cstdio>
#include <stack>
#include <cstring>
using namespace std;
int main(){
    int n, rcd[200];         // rcd 用来记录火车进出站次序
    char O1[100], O2[100];   // O1, O2 分别为题目要求的进站次序和出站次序

    stack <char> s;          // 建立一个栈
    while (scanf("%d %s %s", &n, O1, O2) != EOF) {
        memset(rcd, -1, sizeof(rcd));
        int j = 0, k = 0;
        while (!s.empty()) s.pop();
        for (int i = 0; i < n; i++) {
            s.push(O1[i]);
            rcd[k++] = 1;
            while (!s.empty() && s.top() == O2[j]) {
                s.pop();
                rcd[k++] = 0;
                j++;
            }
        }
        if (s.empty()) {
            printf("Yes.\n");
            for (int i = 0; i < 2 * n; i++) {
                if (rcd[i] == 1) printf("in\n");
                else if (rcd[i] == 0) printf("out\n");
            }
            printf("FINISH\n");
        } else {
            printf("No.\nFINISH\n");
        }
    }
    return 0;
}