由两个栈组成的队列
编写一个类,用两个栈实现队列,支持队列的操作(add,poll,peek)
栈是先进后出的,而队列为先进先出的,我们可以先将数据加入到一个栈中,然后再将数据倒入另一个栈,这样正好就可以实现先进先出,例如:数据进Push栈的顺序为1,2,3,4,5,出Push栈的顺序为5,4,3,2,1,然后进入Pop栈顺序为5,4,3,2,1,出Pop栈的顺序为1,2,3,4,5,正好符合队列的性质。不理解的话,画个图就清楚啦。此题就是考察了对栈和队列的理解。
有两个容易出错的地方:
- 等Pop栈中没有数据后在将Push栈中的全部压入Pop栈。
- 压入数据时必须将Push栈中的数据全部压入到Pop栈。
public class QueueInTowStack {
private Stack<Integer> stackPush = new Stack<>();
private Stack<Integer> stackPop = new Stack<>();
public void add(Integer date){
stackPush.push(date);
}
public Integer poll() throws Exception {
if(stackPush.isEmpty() && stackPop.isEmpty())
throw new Exception("Queue isEmpty!");
pushToPop();
return stackPop.pop();
}
public Integer peek() throws Exception {
if(stackPush.isEmpty() && stackPop.isEmpty())
throw new Exception("Queue isEmpty!");
pushToPop();
return stackPop.peek();
}
private void pushToPop(){
if(stackPop.isEmpty()){
while(!stackPush.isEmpty()){
stackPop.push(stackPush.pop());
}
}
}
}