UVA514 Rails （stack）

题目链接：https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=455

题意：从A方向来一个1,2,3,4.....N顺序的列车，经过车站，输入一个从B出来的列车的编号a1,a2,a3,a4....an，问从B出来的列车编号是否合法。对于车站来说，只能够从A进从B出，每次只能够有一个操作（从A进或者从B出，还可以待在车站）；

思路：用栈来模拟；

 

详细看下面代码：

 

#include <cstdio>
#include <stack>
#include <cstring>
#include <algorithm>
using namespace std;

int n,a[1100],vis[1100];
stack<int > sa;

int main(){
    while(~scanf("%d",&n)){
        while(~scanf("%d",&a[1])){
        if(!a[1]){
            printf("\n");
            break;
        }
        while(!sa.empty()) sa.pop();
        for(int i = 2; i <= n; i ++) scanf("%d",&a[i]);

        memset(vis,0,sizeof(vis));
        int cnt = 0, k = 1;
        for(int i = 1; i <= n && k <= n;){//判断第一个与第一个出来的车相等的编号
            if(a[i] == k){
                vis[k] = 1;
                cnt ++,k ++, i ++;
                while(!sa.empty() && a[i] == sa.top()){//第一个车出车站后，判断车站内的其他车是否是第2,3个出车站的
                    sa.pop(); i ++;
                }
                continue;
            }
            while(a[i] != k && k <= n){//编号为k的车不该出车站则要停留在车站内，即入栈操作
                sa.push(k); k ++;
            }
        }
        k = 1;
        while(!sa.empty()){//判断车站内剩的其他车是否会和B出站的车的编号顺序一样
            if(!vis[a[k]] && sa.top() == a[k])sa.pop(),k ++;
            else if(vis[a[k]]) k ++;
            else break;
        }
        if(sa.empty()) printf("Yes\n");//栈空则表示车站内的车全部从B出去
        else printf("No\n");
        }
    }
    return 0;
}