UVa 514 Rails(栈)

来源:互联网 发布:正商集团怎么样知乎 编辑:程序博客网 时间:2024/05/16 23:37
题目链接:(http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&category=7&page=show_problem&problem=455)

题目描述:

某城市有一个火车站,有n节车厢从A方向驶入车站,按进站的顺序编号为1-n.你的任务是判断是否能让它们按照某种特定的顺序进入B方向的铁轨并驶入车站。例如,出栈顺序(5 4 1 2 3)是不可能的,但是(5 4 3 2 1)是可能的。

题目分析:
为了重组车厢,借助中转站,对于每个车厢,一旦从A移入C就不能回到A了,一旦从C移入B,就不能回到C了,意思就是A->C和C->B。而且在中转站C中,车厢符合后进先出的原则。故这里可以看做为一个栈。

这里写图片描述

解题思路:
当原数组元素等于目标数组元素时,先进栈再出栈即可,否则判断栈顶元素是否等于目标数组,若相等栈顶元素出栈,若不相等在数组不为空的情况下将原数组元素入栈。若都不满足说明不符合条件,flag = 0,break。

#include <cstdio>#include <stack>using namespace std;int N;int target[1005];int main(){    while(~scanf("%d",&N) && N)    {        while(~scanf("%d",&target[1]) && target[1])        {            for(int i = 2;i <= N;i++)                scanf("%d",&target[i]);            int A = 1,B = 1;//A为原数组元素,同时也是下标,B为目标数组下标            int flag = 1;            stack<int> s;            while(B <= N)            {                if(A == target[B])                {                    A++;                    B++;                }                else if(!s.empty() && s.top() == target[B])                {                    s.pop();                    B++;                }                else if(A <= N)                {                    s.push(A);                    A++;                }                else                {                    flag = 0;                    break;                }            }            printf("%s\n",flag?"Yes":"No");        }        printf("\n");    }    return 0;}


0 0
原创粉丝点击