UVA-514Rails(栈)

题目来源:https://vjudge.net/problem/UVA-514

题意:某城市有一个火车站,有n节车厢从A方向驶入车站,按进站顺序编号为1~n。判断是否能让他们按照某种特定的顺序进入B方向并驶出车站。(只有A->C、C->B两种行驶选择)。

UVA-514Rails(栈)_第1张图片

分析:在中转站C中,符合后进先出的规则,所以是一个栈。

参考代码:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include

using namespace std;
const int maxn = 1000+10;
int n;
int a[maxn];

int main()
{
    while( ~scanf("%d",&n) && n)
    {
        while( true)
        {
            scanf("%d",&a[1]);
            if( a[1] == 0)
                break;
            for( int i = 2; i <= n; i++)
                scanf("%d",&a[i]);

            stack s;
            int p = 1;//进栈的~
            int q = 1;//下标
            int flag = 1;
            while( q <= n)//保证下标的合理性
            {
                if( p == a[q])//当前进栈的**和当前数组第q个相等
                {
                    p++;
                    q++;
                }
                else if( !s.empty() && s.top() == a[q])//栈不空并且栈尾元素等于第p个元素
                {
                    s.pop();
                    q++;
                }
                else if( p <= n)
                    s.push(p++);
                else
                {
                    flag = 0;
                    break;
                }
            }
            if( flag)
                printf("Yes\n");
            else
                printf("No\n");
        }
        putchar(10);
    }

    return 0;
}


你可能感兴趣的:(数据结构--基础)