uva514-铁轨rails-题解

题干

某城市有一个火车站，铁轨铺设如图。有n节车厢从A方向驶入车站，按进站的顺序编号为1～n。你的任务是判断是否能让他们按照某种特定的顺序进入B方向的铁轨并驶出车站。例如，出栈顺序（5 4 1 2 3）是不可能的，但（5 4 3 2 1）是可能的。 为了重组车厢，你可以借助中转站C。这是一个可以停放任意多节车厢的车站，但由于末端封顶，驶入C的车厢必须按照相反的顺序驶出C。对于每节车厢，一旦从A移入C，就不能返回A了;一旦从C移入B，就不能返回C了。也就是说，在任意时刻，只有两种选择：A到C和C到B。

对于每一组数据，第一行是一个整数 N 。接下来若干行数据，每行 N 个数，代表 1 ~ N 车厢的出栈顺序，最后一组数据只有一个整数 0 。对于每一组数据，在最后输出空行。

最后一组数据的 N=0 ，不输出。

n<=1000

代码

/*
先进后出，是栈
我们利用栈实现这样一个算法，假设指定顺序为q[n]，模拟栈为a。
想象一个栈，长得像n个铁皮箱子堆起来
对于一个q[i]而言，要么等于栈顶元素，要么大于栈顶元素，还未入栈
所以我们对q从头开始扫，若是栈为空或者栈顶小于q[i]，则栈顶元素到q[i]之间得元素入栈
若是小于栈顶，结束
若是等于栈顶，出栈一个
*/
#include<iostream>
#include<stack>
using namespace std;
stack <int> all;
bool judge(int *q,int n);
int main(){
    int n;
    while(scanf("%d",&n)&&n!=0){
    int arr[n];
    while(1){
    cin>>arr[0];
    if(!arr[0]){
        cout<<"\n";
        break;
    }
    else{
    for(int i=1;i<n;i++)cin>>arr[i];
    cout<<(judge(arr,n)?"Yes\n":"No\n");
    }
}
}
return 0;
    }
bool judge(int *q,int n){
    while(!all.empty())all.pop();
    int i=0,maxn=0;
//maxn记录已出栈的数的最大值，方便下一次出栈
    for(;i<n-1;i++){
//如果前面都对，那么最后一个数一定是对的，(n个数弹出了n-1个相同的，最后一个也相同)所以i遍历到n-2即可
        if(all.empty()||q[i]>all.top()){
            for(int k=maxn;k<q[i]-1;k++)all.push(k+1);
            maxn=q[i];
        }
        else if(q[i]<all.top())return 0;
        else if(q[i]==all.top())all.pop();
    }
    return 1;
}