牛客网剑指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();
};

猜你喜欢

转载自blog.csdn.net/qq_36132127/article/details/80191706