剑指offer之面试题22:栈的压入、弹出序列

剑指offer之面试题22:栈的压入、弹出序列

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

思路1:

剑指offer之面试题22:栈的压入、弹出序列

代码:

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

将思想精简,代码精简后

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