剑指offer 5.栈和队列 用两个栈实现队列

题目描述

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

 

解题思路：
1，整体思路是元素先依次进入栈1，再从栈1依次弹出到栈2，然后弹出栈2顶部的元素，整个过程就是一个队列的先进先出
2，但是在交换元素的时候需要判断两个栈的元素情况：
“进队列时”，队列中是还还有元素，若有，说明栈2中的元素不为空，此时就先将栈2的元素倒回到栈1 中，保持在“进队列状态”。
“出队列时”，将栈1的元素全部弹到栈2中，保持在“出队列状态”。
所以要做的判断是，进时，栈2是否为空，不为空，则栈2元素倒回到栈1，出时，将栈1元素全部弹到栈2中，直到栈1为空。

 

代码实现：

import java.util.Stack;

public class StackToQueue {

Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();

public void push(int node) {
stack1.push(node);
}

public int pop() {
if (stack1.isEmpty()&&stack2.isEmpty()) {
throw new RuntimeException("Queue is empty!");
}else {
if(stack2.empty()){
while(!stack1.empty()){
stack2.push(stack1.pop());
}
}
}

return stack2.pop();

}
}