剑指Offer——栈的压入、弹出序列

题目描述

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

解题思路

• 需要借助一个辅助栈，i，j分别为pushV和popV当前值的下标，依次对需要压进栈的值和弹出值进行判断：
• 如果两者相等，说明需要将该值压入后再弹出，则i++，j++即可；
• 如果两者不相等，则判断popV[j]和当前栈中栈顶元素是否相等，相等的话将栈顶元素弹出；不相等的话，将pushV[i]压入栈；
• 如果pushV中的值已经遍历完成，则只需要判断popV[j]和当前栈中栈顶元素是否相等，相等的话将栈顶元素弹出；不相等的话，说明该序列不是对应的弹出序列，return false；
• 直到popV中的值也遍历完成，说明该序列是对应的弹出序列，return true。

代码

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

剑指offer思路

和上述思路类似，但是更有条理更容易理解一些，同样需要建立一个辅助栈，具体步骤如下：

（1）如果下一个弹出的数字刚好是栈顶数字，那么直接弹出；

（2）如果一下个弹出的数字不在栈顶，则把压栈序列中还没有入栈的数字压入辅助栈，直到把下一个需要弹出的数字压入栈顶；

（3）如果所有数字都压入栈仍然没有找到下一个弹出的数字，那么该序列不可能是一个弹出序列。

代码

class Solution {
public:
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        stack<int> st;
        int id=0;
        for(int i=0;i<popV.size();++i){
            while(st.empty()||st.top()!=popV[i]){
                st.push(pushV[id++]);
                if(id>pushV.size()){
                    return false;
                }
            }
            st.pop();
        }
        return st.empty();
};