1077: 铁轨(PIPIOJ)

1077: 铁轨(PIPIOJ)

题目链接

题目描述

PopPush城市有一座著名的火车站。这个国家到处都是丘陵。而这个火车站是建于上一个世纪。不幸的是，那时的资金有限。所以只能建立起一条路面铁轨。而且，这导致这个火车站在同一个时刻只能一个轨道投入使用，因为它缺少空间，两列火车将无路可走。具体看下图。
当地的惯例是每一列火车从A方向驶向B方向时候，会用某种方式将车厢重组。假设火车将要到达A方向，拥有N个车厢(N<=1000),这些车厢按照递增顺序标记为1到N。负责重组车厢的领导,必须知道是否能重组车厢让它驶出B，而这个重组的序列就是a1\a2\a3…aN.请帮助他写一个程序来判断是否可能按照所要求的车厢顺序。你可以假设，单个的车厢可以从列车上分离出来，在他们进入站台之前。并且他们可以自由移动，直到它们上了B轨道。你也可以假设在任意时候站台可以放下无数的车厢。但是只要一个车厢进入站台，它就不能返回A轨道，同时如果它离开了站台驶向B轨道，它就不能返回站台。

输入

输入包含多组测试用例。
对于每组测试用例，第一行包含一个正整数n(1<=n<=1000)。
以下每一行包含n个数字，代表列车的驶出序列。
每组测试用例以一个单独的0结尾。
以一个单独的0作为结束输入。

输出

对于每一组测试样例的每一行，输出驶出顺序是否合法。
测试样例间以空行进行分隔。

样例输入

5
1 2 3 4 5
5 4 1 2 3
0
6
6 5 4 3 2 1
0
0

样例输出

Yes
No

Yes

题目思路

这是到STL-stack板子题，先判断要重组的第一个车厢，是不是Station的顶部的车厢 如果是的，把Station顶部的车厢移到B处再寻找B处要的下一个车厢，如果不是的再判断是不是A处的第一个车厢，如果不是的，把A处的车厢移到Station中，再判断A处的下个车厢是不是B处要的车厢如果是的话移到Station再移到B处，然后依次按照上述方法判断B剩下的车厢。当我们在Station和A处的首部都找不到B处的车厢的话就说明不能组成

这个Station就像是一个栈stack，每次放入Station的都是在顶部。而把Station顶部的车厢移到B处就是POP，把A处的车厢移到Station就是PUSH。

上代码

#include<cstdio>
#include<stack>
using namespace std;
int main()
{
    int n, a, train[1005], tem[1005], flag, j;
    stack<int> t;
    while(scanf("%d", &n) && n != 0)
    {
        while(1)
        {
            flag = j = 0;
            scanf("%d", &a);
            if(!a)  break;
            train[0] = a;
            for(int i = 1; i < n; i++)
            {
                scanf("%d", &train[i]);
                tem[i-1] = i;
            }

            tem[n-1] = n;
            for(int i = 0; i < n; i++)
            {
               if(!t.empty())
               {
                   if(t.top() == train[j])
                   {
                       j++;
                       t.pop();
                       i--;
                       continue;
                   }
               }
               if(tem[i] == train[j])
               {
                   j++;
                   continue;
               }
               else
                    t.push(tem[i]);
            }
            while(!t.empty())
            {
                if(t.top() == train[j])
                {
                    j++;
                    t.pop();
                }
                else
                {
                    flag = 1;
                    break;
                }
            }
            while(!t.empty())
                t.pop();
            if(!flag)    printf("Yes\n");
            else        printf("No\n");
        }
        printf("\n");
    }
    return 0;
}