最近开始看数据结构,也开始刷题,真的太菜了,什么都不会。
题目
用两个栈来实现一个队列,完成队列的 Push 和 Pop 操作。
分析下题目,也就是说,一个栈用来push,一个栈用来pop。而且顺序是先进先出!
每次push一个元素进栈1,这个比较容易理解。
对于pop,思路大概如下:
- 如果栈2中还有元素,那么先把栈2的元素出栈,这一点至关重要。前面没考虑这个问题,导致继续向栈2压入元素,使得出栈是顺序不对。
- 如果栈2为空,那么从栈1出栈的元素分别入栈栈2.
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) {
//栈1负责入栈
stack1.push(node);
}
public int pop() throws Exception {
//栈2负责出栈
if(stack2.isEmpty()){
while (!stack1.isEmpty()){
stack2.push(stack1.pop());
}
}
if (stack2.isEmpty()){
throw new Exception("queue is empty");
}
return stack2.pop();
}