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

原题

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

Reference Answer

解题思路:

新建一个栈，和一个指向出栈序列第一个元素的指针。按照入栈顺序存入序列。当任一时刻栈顶元素和指针指向的出栈序列的第一个元素相同时，弹出该栈顶元素，并右移指针；继续判断新的栈顶元素与指针指向的元素是否相同。当不同时继续按入栈序列将元素入栈，并重复上述步骤。

def IsPopOrder(self, pushV, popV):
    stack = []
    index = 0
    while pushV:
        stack.append(pushV.pop(0))
        while stack:
            if stack[-1] == popV[index]:
                stack.pop(-1)
                index += 1
            else:
                break
    if not stack:
        return True
    else:
        return False

错误答案：

# -*- coding:utf-8 -*-
class Solution:
    def IsPopOrder(self, pushV, popV):
        # write code here
        stack = []
        index = 0
        while pushV and popV:
            stack.append(pushV.pop(0))
            while stack:
                
                if stack[-1] ==  popV.pop(0):
                
                    stack.pop(-1)
                    popV.pop(0)
                    index += 1
                else:
                    break
        if not stack:
            return True
        else:
            return False
                   

错误原因：if stack[-1] == popV.pop(0):，每次比较时候，无论是否满足条件，popV都会减少一个元素，故而不能直接用if stack[-1] == popV.pop(0):做对比，可以用popV[0]，满足条件后再popV.pop(0)

判断语句有点繁杂，优化后的代码如下：

优化后Python

class Solution:
    def IsPopOrder(self, pushV, popV):
        stack = []
        index = 0
        while pushV:
            stack.append(pushV.pop(0))
            while stack and stack[-1] == popV[index]:
                stack.pop(-1)
                index += 1
        return not stack

二刷

思路

• 使用一个辅助栈
• 依次将入栈序列入栈，如果栈顶元素等于出栈序列的栈顶元素，则弹出
• 当流程无法继续时，如果辅助栈是空的，则出栈序列是符合的

Python Version:

# -*- coding:utf-8 -*-
class Solution:
    def IsPopOrder(self, pushV, popV):
        # write code here
        if not pushV or not popV:
            return False
        stack = []
        j = 0
        for i in range(len(pushV)):
            stack.append(pushV[i])
            while stack and stack[-1] == popV[j]:
                stack.pop()
                j += 1
        return False if stack else True
            

C++ Version:

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

参考答案：https://blog.youkuaiyun.com/u010005281/article/details/79904006