剑指offer:用两个栈实现队列
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:元素先进stack1,再从stack1到stack2,如果stack2中有元素则直接弹出,没有则从stack1中获取元素。当stack2中有元素的时候,stack1不能向stack2添加元素,否则会造成乱序。
进入元素: 1 2 3 4 5
先进入stack1得
stack1
5
4
3
2
1
再从stack1到stack2得
stack2
1
2
3
4
5
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() { if(!stack2.isEmpty()){ //如果stack2不为空,则直接弹出 return stack2.pop(); }else{ while(!stack1.isEmpty()){ stack2.push(stack1.pop()); } } return stack2.pop(); } }