原创

【剑指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
正文到此结束
本文目录