GRYZ 模 拟 赛 系 列 Xxy 的车厢调度

Xxy 的车厢调度
(train.cpp/c/pas)
Description
有 一 个 火 车 站 ， 铁 路 如 图 所 示 ，每辆火车从 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。

思lu:这个栈不会上溢，所以不能成立的原因只有当要出车1的时候上边有车2等阻拦；

Top作为栈顶所以当B数组为当前需要top到达的车的位置，stack[top-1]==0&&top!=b[i])||top-1==b[i]时top—若top最后==车则本次可以成立不等则判断否return

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<string>
using namespace std;
int b[1003];
int top=1,n;
bool flag=false;
int stack[1003];
void mdzz()
{
    for(int i=1;i<=n;i++)
    {
        if(top<b[i])
        {
            while(top != b[i])
            {
             top++;
            }
        stack[b[i]]=0;    
        }
        else if(top>b[i])
        {
            while((stack[top-1]==0&&top!=b[i])||top-1==b[i])
            {
                top--;
            }
            if(top!=b[i])
            {
                flag=true;
                return;
            }
            else stack[b[i]]=0;
        }
        
    }
}
int main()
{
    freopen("train.in","r",stdin);
    freopen("train.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        stack[i]=1;
        scanf("%d",&b[i]);
    }
    mdzz();
    if(flag==1)printf("NO");
    else printf("YES");
    fclose(stdin);
    fclose(stdout);
    return 0;
}