栈的应用——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 =  + ;
 int n, target[maxn];

 int main()
 {
     while (scanf("%d",&n) == )
     {
         stack<int>s;
         int A = , B = ;
         for (int i = ; 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 ;
 }