UVa 514 Rails(栈)

题目链接：(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;
}