【剑指Offer】021——栈的压入、弹出序列 （栈）

题目描述

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

解题思路

模拟堆栈操作的过程，将原数列依次压栈，把栈顶元素与所给出栈队列相比，如果相同则出栈，如果不同则继续压栈，直到原数列中所有数字压栈完毕。最后，检测栈中是否为空，若空，说明出栈队列可由原数列进行栈操作得到。否则，说明出栈队列不能由原数列进行栈操作得到。（根据序列数据压栈、出栈）

参考代码

Java

import java.util.ArrayList;
import java.util.Stack;

public class Solution {
    public boolean IsPopOrder(int[] pushA, int[] popA) {
        if (pushA.length != popA.length || pushA.length == 0 || popA.length == 0)
            return false;
        Stack<Integer> stack = new Stack<Integer>();
        int index = 0;  // popA的索引
        for (int i = 0; i < pushA.length; i++) {
            stack.push(pushA[i]); // 压栈
            // 每压一个值就比较
            while (!stack.empty() && stack.peek() == popA[index]) {
                stack.pop(); // 弹出
                index++;
            }
        }
        return stack.empty();
    }
}

Python

# -*- coding:utf-8 -*-
class Solution:
    def IsPopOrder(self, pushV, popV):
        if len(pushV) != len(popV) or len(pushV) == 0 or len(popV) == 0: return False
        helper = []
        index = 0       # popv的索引
        for i in range(len(pushV)):
            helper.append(pushV[i])
            while helper and helper[-1] == popV[index]:
                helper.pop() # 比较一个就弹出一个
                index += 1
        return True if len(helper) == 0 else False

个人订阅号

image