铁轨

试题描述

某城市有一个火车站，铁轨铺设如图所示，有n节车厢从A方向驶入车站，按进站顺序编号为1至n。你的任务是判断是否能让它们按照某种特定的顺序进入B方向的铁轨并驶出车站。为了重组车厢，你可以借助中转站C。这是一个可以停放任意多节车厢的车站，但由于末端封顶，驶入C的车厢必须按照相反的顺序驶出车站。例如：出站顺序（5，4，1，2，3）是不可能的，而（5，4，3，2，1）是可能的。对于每节车厢，一旦从A移入C，就不能再回到A了；一旦从C移入B，就不能回到C了。换句话说，在任意时刻，只有两种选择A->C和C->B。                        现在给你一种1到n的排列，请你判断是否是题目描述的一种可能，如果是请输出yes，否则输出no

输入

两行，第一行有一个正整数n，表示有n节车厢，第二行有n个正整数，即1到n的一种排列，两两之间有一个空格分隔。

输出

yes或者no

输入示例

5 5 4 1 2 3

输出示例

no

其他说明

对于 100%的数据，0 < n ≤ 100。

 

#include <iostream>
#include <stack>
#include <cstdio>
#include <cstring>
#define MAXN 100 + 10
using namespace std;
int a[MAXN],n;
int main()
{
    memset(a,66,sizeof(a));
    stack<int> s;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    int A=1,B=1;
    bool book=true;
    while(B<=n)
    {
        if(A==a[B])
        {
            A++;
            B++;
        }
        else if(!s.empty() && s.top()==a[B])
        {
             s.pop();
             B++;
        }
        else if(A<=n)
             s.push(A++);
        else
        {
            book=false;
            break;
        }
    }
    if(book)
        printf("yes");
    else
        printf("no");
    //system("pause");
}