由两个栈组成的队列

题目

编写一个类，用两个栈实现队列，支持队列的基本操作（add, poll, peek)

要求

无

思路

使用两个栈，栈A用于add，栈B用于poll和peek，add的时候，往栈A中push元素，poll时，如果栈B不是空，则弹出最顶元素，如果为空则将栈A的所有元素pop，然后push到栈B中。这样就可以实现队列的功能了。

代码

package com.github.zhanyongzhi.interview.algorithm.stacklist;

import java.util.Stack;

/**
* 用两个栈实现队列
* @author zhanyongzhi
*/
public class TwoStackQueue {
private Stack pushStack = new Stack();
private Stack popStack = new Stack();

public T add(T item){
pushStack.push(item);
return item;
}

public T poll(){
if(popStack.empty()){
moveStack();
}

if(!popStack.empty())
return popStack.pop();

return null;
}

public T peek(){
if(popStack.empty()){
moveStack();
}

if(!popStack.empty())
return popStack.peek();

return null;
}

private void moveStack(){
int count = pushStack.size();
for(int i=0; i<count; i++){
T item = pushStack.pop();
popStack.push(item);
}
}
}

在github中查看