题目描述
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
解题思路
考虑栈和队列的特性,栈为先进后出,队列为先进先出,可以认为栈A提供入队列即push的功能,栈B提供出队列即pop功能。
对于push,直接将数据添加进栈A即可。
对于pop,需要分两种情况考虑:
1)若栈B不为空,则直接弹出栈B的数据。
2)若栈B为空,则先依次弹出栈A的数据,放入栈B中,再弹出栈B的数据。
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.add(node);
}
public int pop() {
if(!stack2.empty()) return stack2.pop();
else{
while(!stack1.empty()){
stack2.push(stack1.pop());
}
return stack2.pop();
}
}
}