首页 > 代码库 > UVA 514 Rails

UVA 514 Rails

UVa514 Rails(铁轨)

题目描述:
某城市有一个火车站,有n节车厢从A方向驶入车站,按进站的顺序编号为1-n.你的任务是判断是否能让它们按照某种特定的顺序进入B方向的铁轨并驶入车站。例如,出栈顺序(5 4 1 2 3)是不可能的,但是(5 4 3 2 1)是可能的。
题目分析:
为了重组车厢,借助中转站,对于每个车厢,一旦从A移入C就不能回到A了,一旦从C移入B,就不能回到C了,意思就是A->C和C->B。而且在中转站C中,车厢符合后进先出的原则。故这里可以看做为一个栈。

#include <iostream>
#include <stack>
using namespace std;

const int MAXN = 1000 + 10;
int n, target[MAXN];

int main()
{
    while(cin >> n && n)
    {
        for(int i = 1; i <= n; i++)
        {
            cin >> target[i];
        }

        stack<int> s;
        int A = 1,B = 1;
        int ok = 1;

        while(B <= n)
        {
            if(A == target[B])
            {
                A++;
                B++;
            }
            else if(!s.empty() && s.top() == target[B])
            {
                s.pop();
                B++;
            }
            else if(A <= n)
                s.push(A++);
            else
            {
                ok = 0;
                break;
            }
        }
        cout << (ok ? "Yes" : "No" )<<endl; //注意这里必须括号括起来
    }
    return 0;
}
本文参考算法竞赛入门经典(刘汝佳)以及挑战程序设计

UVA 514 Rails