栈的应用——Rails

一、题目描述

某城市有一个火车站，有n节车厢从A方向驶入车站，按进站顺序编号为1~n,经中转站C驶向B。中转站C，这是一个可以停放任意多节车厢的车站，但由于末端封顶，驶入C的车厢必须以相反的顺序驶出C。你的任务是判断它能否按某种顺序进入B方向的车站。

二、解题思路

中转站C就相当于一个栈，可以随时入栈和出栈。

三、代码实现

#include<stdio.h>
#include<iostream>
#include<stack>
#include<cstdbool>
using namespace std;

const int maxn = 100 + 10;
int n, target[maxn];

int main()
{
    while (scanf("%d",&n) == 1)
    {
        stack<int>s;
        int A = 1, B = 1;
        for (int i = 1; i <= n; i++)
            scanf("%d", &target[i]);

        bool ok = true;
        while (B <= n)
        {
            if (A == target[B]){        //从A中取
                A++, B++; 
            }
            else if (!s.empty() && s.top() == target[B]) { //从栈中取
                s.pop();
                B++;
            }
            else if (A <= n)  s.push(A++);        //栈顶不是目标元素，从A入栈
            else {                            //都不是，说明不可能
                ok = false;
                break;
            }
        }
        printf("%s\n", ok ? "Yes" : "No");
    }
    return 0;
}