劍指offer：棧的壓入、彈出序列



題目描述：

輸入兩個整數序列，第一個序列表示棧的壓入順序，請判斷第二個序列是否爲該棧的彈出順序。假設壓入棧的所有數字均不相等。例如序列1,2,3,4,5是某棧的壓入順序，序列4,5,3,2,1是該壓棧序列對應的一個彈出序列，但4,3,5,1,2就不可能是該壓棧序列的彈出序列

這道題考察的是STL中的棧，與ZOJ上的火車進站問題一樣。

class Solution {
public:
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        int len1 = pushV.size(), len2 = popV.size();
        if(len1 == 0 || len2 == 0)
            return false;
        if(len1 == 1 && len2 == 1)
            {
            if(pushV[0] == popV[0])
                return true;
            else
                return false;
        }
        if(len1 != len2)
            return false;
        stack<int> s;
        int i = 0, j = 0;
        while(i < len1 && j < len1)
            {
            if(s.empty() || pushV[i] != popV[j])
            {
            s.push(pushV[i]);
            i++;
        }
            if(pushV[i] == popV[j])
                {
                i++;
                j++;
                while(!s.empty() && popV[j] == s.top() && j < len1)
                    {
                    s.pop();
                    j++;
                }
            }
        }
        if(j == len1)
            return true;
        else
            return false;
    }
};