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

题目描述

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

 

我的代码

class Solution {
public:
    int i = 0, j = 0;
    vector<int> temp;
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        while(i < pushV.size() && j < popV.size())
        {
            while((temp.empty() || temp.back() != popV[j]) && i < pushV.size())
            {
                temp.push_back(pushV[i++]);
            }
            else if(temp.back() == popV[j])
            {
                temp.pop_back();
                j++;
            }
            else if(pushV[i] == popV[j])
            {
                i++;
                j++;
            }
        }
         
        if(i == pushV.size())
        {
            while(!temp.empty() && j < popV.size())
            {
                int value = temp.back();
                temp.pop_back();
                if(value != popV[j])
                    return false;
                j++;
            }
            if(temp.empty() && j == popV.size())
                return true;
            else
                return false;
        }
    }
};

嗯，乱七八糟，不能通过。

 

别人的代码：

class Solution {
public:
    int i = 0, j = 0;
    vector<int> temp;
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        for(int j = 0; j < popV.size(); j++)
        {
            while(temp.empty() || temp.back() != popV[j])
            {
                temp.push_back(pushV[i++]);
                if(i > pushV.size())
                    return false;
            }
            temp.pop_back();
        }
        if(temp.empty())
            return true;
        else 
            return false;
    }
};

（1）将 temp.back() == popV[j] 与 pushV [i]  == popV [j] 两种情况合为了一种。思路很清晰

（2）无效条件的放置也很精巧，i > pushV.size() 说明 pushV 中的元素少了，if (temp.empty()) 说明 pushV 里的元素多了。

（3）除了vector，这里也用一个 stack 来模拟过程。

（4）这里是从popV的角度来考虑，下面换成pushV。

 

最简洁的写法

class Solution {
public:
    vector<int> temp;
    
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        for(int i = 0, j = 0; i < pushV.size();)
        {
            temp.push_back(pushV[i++]);
            while(j < popV.size() && temp.back() == popV[j])
            {
                temp.pop_back();
                j++;
            }
        }
        return temp.empty();
    }
};