UVA ~ 514 ~ Rails （栈）

最新推荐文章于 2024-11-10 02:14:24 发布

原创最新推荐文章于 2024-11-10 02:14:24 发布 · 238 阅读

0 ·

CC 4.0 BY-SA版权

【栈/队列/优先队列/链表】专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种火车调度问题的算法实现，通过栈的数据结构来模拟中转站的作用，判断是否能够按照特定顺序重新排列车厢。文章提供了完整的C++代码实现，并详细解释了算法流程。

题意：某城市有一个火车站，铁轨铺设如图所示。有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。

分析：在中转站C中，车厢符合后进先出的原则，因此是一个栈。

以上内容来自算法竞赛入门经典。

#include<bits/stdc++.h>
using namespace std;
const int MAXN = 1005;
int n, t[MAXN];
int main()
{
    while (~scanf("%d", &n) && n)
    {
        while (~scanf("%d", &t[1]) && t[1])
        {
            stack<int> S;
            int A = 1, B = 1;
            for (int i = 2; i <= n; i++)
                scanf("%d", &t[i]);
            bool ok = true;
            while (B <= n)
            {
                if (A == t[B]) { A++; B++; }
                else if (!S.empty() && S.top() == t[B]) { S.pop(); B++; }
                else if(A <= n) S.push(A++);
                else { ok = false; break; }
            }
            printf("%s\n", ok ? "Yes" : "No");
        }
        printf("\n");
    }
    return 0;
}