(剑指Offer)面试题22:栈的压入、弹出序列

题目:

输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。

例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。

思路:

直观的想法就是模拟第一个序列的数字依次入栈,并按照第二个序列依次弹出,如果第一个序列全部入栈,并最后栈中元素都被弹出来,就说明第二个序列是该栈的弹出顺序。

  • 建立一个辅助栈;
  • 将第一个序列的数字压入辅助栈;
  • 如果第二个序列的下一个弹出数字刚好是栈顶数字,则直接弹出,第二个序列弹出当前数字,辅助栈也弹出该数字;
  • 否则,就把第一个序列中尚未入栈的数字压入辅助栈,直到把第二个序列中下一个需要弹出的数字压入栈顶为止。
  • 如果第一个序列的所有数字都已经入栈了,仍然没有找到下一个弹出的数字,那么第二个序列不可能是一个弹出序列。

代码:

#include <iostream>
#include <stack>

using namespace std;

bool IsPopOrder(const int* push,const int* pop,int length){
    bool bPossible=false;
    if(push!=NULL && pop!=NULL && length>0){
        std::stack<int> stackData;
        int i=0;
        int j=0;
        while(j<length){
            while(stackData.empty() || stackData.top()!=pop[j]){
                if(i>length-1)
                    break;
                stackData.push(push[i]);
                ++i;
            }
            if(stackData.top()!=pop[j])
                break;
            stackData.pop();
            ++j;
        }
        if(stackData.empty() && j==length)
            bPossible=true;
    }
    return bPossible;
}

int main()
{
    int A[]={1,2,3,4,5};
    int B[]={4,5,3,2,1};
    int length=sizeof(A)/sizeof(A[0]);

    cout << IsPopOrder(A,B,length) << endl;
    return 0;
}

在线测试OJ:

http://www.nowcoder.com/books/coding-interviews/d77d11405cc7470d82554cb392585106?rp=1

AC代码:

class Solution {
public:
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        int length=pushV.size();
        if(pushV.size()>0 && popV.size()>0 && pushV.size()==popV.size()){
            std::stack<int> stackData;
            int i=0;
            int j=0;
            while(j<length){
                while(stackData.empty() || stackData.top()!=popV[j]){
                    if(i>length-1)
                        break;
                    stackData.push(pushV[i]);
                    ++i;
                }
                if(stackData.top()!=popV[j])
                    break;
                stackData.pop();
                ++j;
            }
            if(stackData.empty() && j==length)
                return true;
        }
        return false;
    }
};
原文地址:https://www.cnblogs.com/AndyJee/p/4651281.html