【剑指offer】刷题记录-栈的压入、弹出序列

题目描述

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

思路:通过画图我们可以找到一些规律:如果下一个弹出的元素刚好是栈顶元素,那么直接弹出。如果下一个弹出的元素不在栈顶,我们可以把数据栈中还没有压入辅助栈的元素压入,直到把下一个需要弹出的元素压入栈顶为止。如果把所有的元素都压入辅助栈了还没有找到下一个要弹出的元素,那么就不是弹出序列。

【剑指offer】刷题记录-栈的压入、弹出序列_第1张图片

import java.util.Stack;

public class Solution {
	public boolean IsPopOrder(int[] pushA, int[] popA) {
		if (pushA.length == 0 || popA.length == 0) {
			return false;
		}
		Stack stack = new Stack();
		int j = 0;
		// pushA={1,2,3,4,5} popA={4,5,3,2,1}
		for (int i = 0; i < popA.length; i++) {
			// 往stack里压入pushA的元素
			stack.push(pushA[i]);
			// 相同就弹出
			while (j < popA.length && popA[j] == stack.peek()) {
				stack.pop();
				j++;
			}
		}
		return stack.empty() == true;
	}
}

 

你可能感兴趣的:(【剑指offer】刷题记录-栈的压入、弹出序列)