剑指offer系列21之栈的压入、弹出序列

要求

• 时间限制：1秒
• 空间限制：32768K
• 热度指数：271748
• 本题知识点： 栈

题目描述

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

思路

借助了一个辅助栈来模拟入栈和出栈的过程，过程结束后若辅助栈为空则满足条件。时间复杂度为$O(n^2)$，空间复杂度为$O(n)$。

python实现

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

• 运行时间：30ms

• 占用内存：5644k

C++实现

class Solution {
public:
    bool IsPopOrder(vector<int> pushV,vector<int> popV) {
        if(pushV.size() == 0) return false;
        vector<int> stack;
        for(int i = 0,j = 0 ;i < pushV.size();i++){
            stack.push_back(pushV[i]);
            while(j < popV.size() && stack.back() == popV[j]){ // 判断条件顺序很重要，不然会报错
                stack.pop_back();
                j++;
            }      
        }
        return stack.empty();
    }
};

• 运行时间：4ms
• 占用内存：460k

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

• 运行时间：3ms
• 占用内存：480k