Xxy 的车厢调度

有 一 个 火 车 站 ， 铁 路 如 图 所 示 ，每辆火车从 A 驶入，

再从 B 方向驶出，同时它的车厢可以重新组合。假设

从 A 方向驶来的火车有 n 节（n<=1000） ，分别按照顺

序编号为 1，2，3，…，n。假定在进入车站前，每节

车厢之间都不是连着的，并且它们可以自行移动到 B

处的铁轨上。 另外假定车站 C 可以停放任意多节车厢。

但是一旦进入车站 C，它就不能再回到 A 方向的铁轨

上了，并且一旦当它进入 B 方向的铁轨，它就不能再

回到车站 C。

负责车厢调度的 xxy 需要知道能否使它以

a1,a2,…,an 的顺序从 B 方向驶出，请来判断能否得到

指定的车厢顺序。

Input

输入文件的第一行为一个整数 n，其中 n<=1000，表示有 n 节车厢，第二行为 n 个数字，表

示指定的车厢顺序。

Output

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

包含引号） 。还有，xxy 说了 这题 AC 有糖吃。

Example

train.in train.out

5

5 4 3 2 1

YES

Hint

对于 50%的数据，1<=N<=20。

对于 100%的数据，1<=N<=1000。

思路：。

我们只需要判断：

1对于每一个进车站的车，如果他比已经进来的所有的车的编号都要大时，需要将比他所有小的车全部进入车站

2如果需要出去的车恰好是第一辆的车，那么让他出去

3如果需要出去的车的编号小于最大的编号，就输出NO

 

改了n次才AC的代码：

#include<iostream>
#include<cstdio>
using namespace std;
int a[10001];
int stack[10001];
int top=0;
int cur=1;
int main()
{
    //freopen("train.in","r",stdin);
    //freopen("train.out","w",stdout);
    int n;
    int cur=1;
    cin>>n;
    for(int i=1;i<=n;i++)
    cin>>a[i];
    for(int i=1;i<=n;i++)
    {
        
        while(cur<=a[i])
        {
            stack[++top]=cur++;
            //cur++;
        }
        if(stack[top]==a[i])
            --top;
        else
        {
            cout<<"NO";
            return 0;
        }
    }
    printf("YES");
    return 0;
}