（劍指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;
    }
};