信息学奥赛一本通题目解析：1357：车厢调度(train)（栈）

【题目描述】

有一个火车站，铁路如图所示，每辆火车从A驶入，再从B方向驶出，同时它的车厢可以重新组合。假设从A方向驶来的火车有n节（n≤1000），分别按照顺序编号为1，2，3，…，n。假定在进入车站前，每节车厢之间都不是连着的，并且它们可以自行移动到B处的铁轨上。另外假定车站C可以停放任意多节车厢。但是一旦进入车站C，它就不能再回到A方向的铁轨上了，并且一旦当它进入B方向的铁轨，它就不能再回到车站C。

负责车厢调度的工作人员需要知道能否使它以a1,a2,…,an的顺序从B方向驶出，请来判断能否得到指定的车厢顺序。

【输入】

第一行为一个整数n，其中n≤1000，表示有n节车厢，第二行为n个数字，表示指定的车厢顺序。

【输出】

如果可以得到指定的车厢顺序，则输出一个字符串“YES”，否则输出“NO”（注意要大写，不包含引号）。

【输入样例】

5
5 4 3 2 1

【输出样例】

YES

【提示】

解析：观察发现，整个调度过程其实是在模拟入栈出栈的过程，而这个过程中，我们可以分成三种状态：栈前、栈中、栈后。我们可以发现，当某个数字出栈了，说明比它小的数字要么已经出栈了，要么还在栈里，不能是入栈前状态，并且在栈中的顺序是从大到小的(从栈顶往栈底看)，比如出5，那么1，2，3，4要么已经在5之前出了，要么还在栈中（假如1，3，4在栈中，从栈顶往栈底看依次为4，3，1），不能是入栈前的状态。如果某个数字要出栈，那么当前在栈中的数字都必须小于它，否则就与栈的性质矛盾，不合法，于是我们可以这样解决：

从第一个数字开始扫描，a[i]表示当前出栈的数字，如果有比a[i]大的数字还在栈中，那么就产生矛盾，输出“NO”；否则，标记当前数字a[i]为栈后状态，那么[1, a[i]-1]这些数字如果还没出栈，标记为栈中状态。具体我们可以用0表示为确定状态，1表示栈中状态，2表示栈后状态。

【解题思路】

思路分为两个主要部分，进栈和出栈，以下是详细的分析：

进栈过程

在进栈过程中，主要关注将车厢按照需要的顺序准备好，以便能够按目标序列正确地从栈中移出。这一过程包括：

• 监测和填充栈：当当前栈顶车厢不是想要驶出的下一节车厢时，继续将从当前位置到目标车厢的所有车厢依次推入栈中。这确保了当需要某节车厢时，它已经在栈顶可供输出。

出栈过程

在出栈过程中，核心操作是按照目标序列要求从栈中取出车厢。这包括：

• 验证和输出：每次循环，检查栈顶车厢是否是目标序列中下一节需要输出的车厢。如果是，从栈中弹出；如果不是，但栈已无法继续填充新的车厢，说明无法达到目标顺序，程序输出“NO”并终止。

代码的合理性

• 进栈：通过 while 循环，将当前需要的车厢推入栈中直到栈顶车厢是目标车厢 x。
• 出栈：在 while 循环之后立即检查栈顶元素。如果栈顶元素是所需的车厢 x，则从栈中弹出。

【代码实现】

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

int a[1005]; // 数组a用于存储车厢的原始顺序，即1到n

int main() {
    stack<int> s; // 使用栈s来模拟车站C的临时存放功能
    int n; // n表示车厢的数量
    int cur = 1; // cur跟踪下一个应该进栈的车厢编号
    cin >> n; // 输入车厢数量
    for (int i = 1; i <= n; i++) {
        a[i] = i; // 初始化车厢的原始顺序
    }

    for (int i = 1; i <= n; i++) {
        int x;
        cin >> x; // 读入目标序列中的车厢编号
        
        // 入栈：确保所有未处理的车厢在需要时可以进栈
        // 如果栈为空或栈顶不是目标车厢，并且还有车厢未进栈，将它们推入栈中
        while (s.empty() || (!s.empty() && s.top() != x && cur <= n)) {
            s.push(a[cur]); // 将车厢推入栈中
            cur++; // 更新下一个应该进栈的车厢编号
        }
        
        // 出栈：如果栈顶车厢是目标序列中的当前车厢，弹出栈
        if (!s.empty() && x == s.top()) {
            s.pop(); // 弹出栈顶车厢，表示该车厢已正确驶出
        } else {
            // 如果无法弹出需要的车厢，输出NO并结束程序
            cout << "NO";
            return 0;
        }
    }
    
    // 如果所有车厢都能按照目标序列顺利处理，输出YES
    cout << "YES";
    return 0;
}