【题解】车厢调度

题目描述

        有一个火车站，铁路如图所示，每辆火车从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个数字，表示车辆从B方向驶出的车厢顺序。

输出格式

        如果工作人员可以调到得到指定的车厢顺序，则输出一个字符串”YES”，否则输出”NO”。

输入输出样例

输入样例

5

5 4 3 2 1

扫描二维码关注公众号，回复： 5155077 查看本文章

输出样例

YES

 

题解

        题目中描述的车厢调度的过程，其实就符合先进后出的特性，和栈的结构相似，我们可以直接模拟栈来判断。

#include<iostream>
using namespace std;
int n,a[1005],top,b[1005],num;
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=1;i<=n;i++)
    {
        while(b[top]!=a[i])
        {
            if(num<n)
            {
                num++;
                top++;
                b[top]=num;
            }
            else
            {
                ;cout<<"NO";
                return 0;
            }
        }
        b[top]=0;
        top--;
    }
    cout<<"YES";
    return 0;
}

参考程序