UVA 514 Rails(模擬|棧)

題目鏈接

題意

n個火車進站，給出一個給定的n個火車的出站順序，問是否可行

解決

棧模擬，主要分三種情況
1. 一進棧就直接出棧
2. 目標編號和棧頂元素相同，棧頂元素出棧

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int num[1005];
    while(1)
    {
        int n;
        scanf("%d",&n);
        if(!n) break;
        while(1)
        {
            scanf("%d",&num[1]);
            if(!num[1]) break;

            for(int i=2;i<=n;i++) scanf("%d",&num[i]);

            int a=1,b=1;                //a指針指向初始入站順序 1,2,3,4....n  b指針指向目標順序
            bool error = false;
            stack<int> S;

            while(b <= n)
            {
                if( a == num[b]){       //這種情況是剛入站，就出站的情況
                    a++;
                    b++;
                }
                else if(!S.empty() && num[b] == S.top()){       //這種情況是棧頂元素和目標元素相同，棧頂元素出棧
                    S.pop();
                    b++;
                }
                else if(a < n){                         //否則  a元素入棧
                    S.push(a);
                    a++;
                }
                else{
                    error = true;
                    break;
                }
            }

            if(error) puts("No");
            else puts("Yes");
        }
        puts("");
    }
}