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("");
    }
}