【用两个栈实现队列】

题目描述

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。


解法

这题相对而言比较简单,栈的特点是先进后出,队列的特点是先进先出,因此用两个先进后出的栈就能实现先进先出的队列,在push时直接往stack1push,在pop时,借助第二个栈暂存stack1的其它元素,让stack1栈底的元素(即最先进来的元素)出栈,出栈后不需要立刻将stack2的元素弹回stack1,因为如果下个操作还是pop的话,就可以直接从stack2弹出。

import java.util.Stack;

public class Solution {
	Stack<Integer> stack1 = new Stack<Integer>();
	Stack<Integer> stack2 = new Stack<Integer>();

	public void push(int node) {
		while (!stack2.isEmpty())
			// s2不空,先将所有值转到s1
			stack1.add(stack2.pop());
		stack1.add(node);
	}

	public int pop() {
		while (!stack1.isEmpty())
			// s2为空
			stack2.add(stack1.pop());
		return stack2.pop();
	}
}

你可能感兴趣的:(剑指offer)