22. 栈的压入、弹出序列

题目描述

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序，序列4，5,3,2,1是该压栈序列对应的一个弹出序列，但4,3,5,1,2就不可能是该压栈序列的弹出序列。

解析

如果下一个弹出的数字刚好是栈顶数字，那么直接弹出。如果下一个弹出的数字不在栈顶，需要把压栈序列中还没有入栈的数字压入辅助栈，直到把下一个需要弹出的数字压入栈顶为止。如果所有数字都压入栈了，仍然没有找到下一个弹出的数字，那么该序列不可能是一个弹出序列。

实现

bool IsPopOrder(const int* pPush, const int* pPop, int nLength){
    bool result = false;
    if (pPush == NULL || pPop == NULL || nLength <= 0)
        return result;
    const int* pNextPop = pPop;
    const int* pNextPush = pPush;
    std::stack<int> stackData;  //辅助栈
    while (pNextPop - pPop != nLength){ //循环终止条件：所有弹出序列的数据都从辅助栈中弹出
        //压栈序列入栈
        while (stackData.empty() || stackData.top() != *pNextPop){ //辅助栈为空 或辅助栈顶数据不等于弹出序列要弹出的数据
            if (pNextPush-pPush == nLength)   //压栈序列的数据都压入到辅助栈中
                break;
            stackData.push(*pNextPush);
            pNextPush++;
        }

        if (stackData.top() != *pNextPop)  
            break;
        stackData.pop();
        pNextPop++;
    }
    if (stackData.empty() && pNextPop - pPop == nLength)
        result = true;
    return result;
}