栈acm应用

Problem:

zoj 1259
火车有n节车厢，顺序编号为1,2,3,...,n，从A驶入，从B驶出，车站里能停放任意多节车厢。一旦进入车站就不再回到A方向的铁轨上，一旦进入B方向铁轨，不再回车站。判断一个指定车厢顺序能否从B方向驶出。

Sample Input

5
1 2 3 4 5

5 4 1 2 3

Sample Output

Yes

入火与出火车，根据出的顺序来比较入的顺序

#include<stdio.h>
#include <iostream>
#include <stack>

using namespace std;
int isPopSequence(int input[],int target[],int n)
{
    int A=0,B=0;
    stack<int>s;
    int ok=1;
    while(B<n)
    {
        if(input[A]==target[B])
        {
            A++;
            B++;
        }
        //栈不为空，判断栈顶
        else if(!s.empty() && s.top()==target[B])
        {
            s.pop();//已经相当于A++
            B++; //所以B要加1
        }
        else if(A<n)
        {
            s.push(A++);//用input的下一个来比较
        }
        else
        {
            ok=0;
            break;
        }
    }
    return 1;
}


int main()
{
    int a[5]={1,2,3,4,5};
    int b[5]={2,5,4,3,1};
    if(isPopSequence(a,b,5) == 1)
      cout<<"True"<<endl;
    else
      cout<<"False"<<endl;

    return 0;
}

hdu 1022 Train problem I(栈~~)

题意：

给定俩个个N长的串，第一个是源，第二个是目标。问能否得到目标序列，能的话再输出序列的进出情况。