题目描述
用两个栈实现队列
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
代码实现:
/**
*
* 用两个栈实现队列
*
* 用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
*
* @author HeMing
*
*/
public class QueueByStack {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
public void push(int node) {
stack1.push(node);
}
/**
* 个人实现方法,看了大神的代码,简直不堪入目
*
* 每次出队列都把stack1中的所有元素压入stack2,出栈后再压回stack1
* @return
*/
public int pop() {
if (stack1.empty()) {
throw new RuntimeException("该队列并没有元素!");
}
while(!stack1.empty()){
stack2.push(stack1.pop());
}
int popInt = stack2.pop();
while(!stack2.empty()){
stack1.push(stack2.pop());
}
return popInt;
}
/**
* 左程云的《程序员代码面试指南》的答案
* @return
*/
public int pop1() {
if (stack1.empty()&&stack2.empty()) {
throw new RuntimeException("The queue is empty!");
}
if (stack2.empty()){
while (!stack1.empty()) {
stack2.push(stack1.pop());
}
}//如果stack2为空,就将当前stack1中所有的元素全部填入到stack2中,然后出栈顶元素。
return stack2.pop();
}
}
总结:
- 错误判定要全面,如果队列中没有元素,返回RuntimeException运行时异常信息提示,exception类型简介见博文:http://blog.csdn.net/csdnofheming/article/details/74353759
- 栈的理解不够透彻,该题中没有必要每次出队都要将全部元素出栈到另一个栈,出队后再将剩余元素填入原来的栈。