欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

UVa514-Rails-铁轨--栈的使用

程序员文章站 2024-03-20 20:39:58
...

说明:本题只是主要思想代码,没有按照OJ上的输入输出格式来。要提交OJ的修改下输入即可。

题目:https://vjudge.net/problem/UVA-514


某城市有一个火车站,铁轨铺设如图6-1所示。有n节车厢从A方向驶入车站,按进站顺序编号为1~n。你的任务是判断是否能让它们按照某种特定的顺序进入B方向的铁轨并驶出车站。例如,出栈顺序(5 4 1 2 3)是不可能的,但(5 4 3 2 1)是可能的。

UVa514-Rails-铁轨--栈的使用

为了重组车厢,你可以借助中转站C。这是一个可以停放任意多节车厢的车站,但由于末端封顶,驶入C的车厢必须按照相反的顺序驶出C。对于每个车厢,一旦从A移入C,就不能再回到A了;一旦从C移入B,就不能回到C了。换句话说,在任意时刻,只有两种选择:A→C和C→B。

分析:在中转站C中,车厢符合后进先出的原则,因此是一个栈。代码如下:(注意while循环中各种情况的分析)

#include<iostream>
#include<cstdio>
#include<stack>
using namespace std;
const int maxn=1000+10;
int n,target[maxn];
int main(){
    while(cin>>n){
        stack<int> s;
        int A=1,B=1;
        for(int i=1;i<=n;i++) cin>>target[i];
        int ok=1;
        while(B<=n){ //仔细理解、模拟
            if(A==target[B]){A++;B++;}
            else if(!s.empty()&&s.top()==target[B]){B++;s.pop();}
            else if(A<=n) s.push(A++);
            else {ok=0;break;}
        }
        printf("%s\n",ok?"Yes":"No");
    }
    return 0;
}