剑指offer系列源码-栈的压入、弹出序列

题目1366：栈的压入、弹出序列
时间限制：1 秒内存限制：32 兆特殊判题：否提交：2143解决：816
题目描述：
输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序，序列4，5,3,2,1是该压栈序列对应的一个弹出序列，但4,3,5,1,2就不可能是该压栈序列的弹出序列。
输入：
每个测试案例包括3行：
第一行为1个整数n(1<=n<=100000)，表示序列的长度。
第二行包含n个整数，表示栈的压入顺序。
第三行包含n个整数，表示栈的弹出顺序。
输出：
对应每个测试案例，如果第二个序列是第一个序列的弹出序列输出Yes，否则输出No。
样例输入：
5
1 2 3 4 5
4 5 3 2 1
5
1 2 3 4 5
4 3 5 1 2
样例输出：
Yes
No

注意不要使用stack，使用的内存超过题目限制，因为其是动态增长，内存复制，需要很大的内存空间。改为数组模拟栈。

#include <iostream>
#include<stdio.h>
#include<stack>
using namespace std;
bool isPopSequence(int* pPush,int* pPop,int nLength){
    if(pPush==NULL||pPush==NULL||nLength<=0){
        return false;
    }
    int* pPushNext = pPush;
    int* pPopNext = pPop;
    int pNext = 0;
    int pTop = -1;
    int* stackData = new int[nLength+1];
    while(pPopNext-pPop<nLength){
        while(pTop==-1||stackData[pTop]!=*pPopNext){
            stackData[pNext++] = *pPushNext;
            pPushNext++;
            pTop++;
        }
        if(pTop>=0&&stackData[pTop]!=*pPopNext){
            return false;
        }
        pNext--;
        pTop--;
        pPopNext++;
    }
    if(pTop==-1&&pPopNext-pPop==nLength){
        return true;
    }
    return false;
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF){
        int* a = new int[n];
        int* b = new int[n];
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
        }
        for(int i=0;i<n;i++){
            scanf("%d",&b[i]);
        }
        if(isPopSequence(a,b,n)){
            printf("Yes\n");
        }else{
            printf("No\n");
        }
    }

    return 0;
}