《剑指Offer》05用两个栈实现队列
题目:
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:
题目很经典,也很简单,push直接push到stack1,pop的话,先将stack1中的元素弹进stack2,
从stack2中pop出一个元素,接着将stack2中的元素弹进stack1。
import java.util.Stack; public class Solution { Stack<Integer> stack1 = new Stack<Integer>(); Stack<Integer> stack2 = new Stack<Integer>(); public void push(int node) { stack1.push(node); } public int pop() { while(!stack1.isEmpty()) { stack2.push(stack1.pop()); } int res = stack2.pop(); while(!stack2.isEmpty()) { stack1.push(stack2.pop()); } System.out.println(res); return res; } }