Java版剑指offer:用两个栈实现队列
描述
用两个栈来实现一个队列,分别完成在队列尾部插入整数(push)和在队列头部删除整数(pop)的功能。 队列中的元素为int类型。保证操作合法,即保证pop操作时队列内已有元素。
示例:
输入:
[“PSH1”,“PSH2”,“POP”,“POP”]
返回:
1,2
解析:
“PSH1”:代表将1插入队列尾部
“PSH2”:代表将2插入队列尾部
"POP“:代表删除一个元素,先进先出=>返回1
"POP“:代表删除一个元素,先进先出=>返回2
示例1
输入:
[“PSH1”,“PSH2”,“POP”,“POP”]
返回值:
1,2
解题思路:
-
队列的特性是:“先入先出”,栈的特性是:“先入后出”
-
当我们向模拟的队列插入数 a,b,c 时,假设插入的是 stack1,此时的栈情况为:
-
栈 stack1:{a,b,c}
-
栈 stack2:{}
-
当需要弹出一个数,根据队列的"先进先出"原则,a 先进入,则 a 应该先弹出。但是此时 a 在 stack1 的最下面,将 stack1 中全部元素逐个弹出压入 stack2,现在可以正确的从 stack2 中弹出 a,此时的栈情况为:
-
栈 stack1:{}
-
栈 stack2:{c,b}
-
继续弹出一个数,b 比 c 先进入"队列",b 弹出,注意此时 b 在 stack2 的栈顶,可直接弹出,此时的栈情况为:
-
栈 stack1:{}
-
栈 stack2:{c}
-
此时向模拟队列插入一个数 d,还是插入 stack1,此时的栈情况为:
-
栈 stack1:{d}
-
栈 stack2:{c}
-
弹出一个数,c 比 d 先进入,c 弹出,注意此时 c 在 stack2 的栈顶,可直接弹出,此时的栈情况为:
-
栈 stack1:{d}
-
栈 stack2:{c}
根据上述栗子可得出结论:
- 当插入时,直接插入 stack1
- 当弹出时,当 stack2 不为空,弹出 stack2 栈顶元素,如果 stack2 为空,将 stack1 中的全部数逐个出栈入栈 stack2,再弹出 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) {
stack1.push(node);
}
public int pop() {
if(stack2.size()<=0){
while(stack1.size()!=0){
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}