Java 使用栈实现简单队列功能
前两天面试奇安信,有问到如果通过栈实现队列,当时没有回答清楚,现在记录一下。
通过A,B两个栈,一个作为入栈,一个作为出栈。当有数据入列时将数据放入A栈,如果需要出列,则调用B栈进行出栈。
具体实现方案
- 当有数据入队时,使用A栈进行入栈,并判断B栈是否为空,如果为空则将A栈的数据转移至B栈
- 当出栈时,判断B栈是否存在数据,如果存在直接弹出,否则转移数据
注意:一定是要在B栈为空时进行转移,否则会导致顺序错乱
public class StackQuene {
//用于入栈
private Stack<Integer> statckPush;
// 用于出栈
private Stack<Integer> stackPop;
public StackQuene(){
stackPop = new Stack<>();
statckPush = new Stack<>();
}
public int poll(){
if(stackPop.empty()){
// 当出栈中没有数据,则进行转移
transferStack();
}
if(stackPop.empty()){
// 数据转移后依然没有数据
return -1;
}
return stackPop.pop();
}
public void add(int num){
statckPush.push(num);
// 如果出栈为空,则进行转移
if(stackPop.empty()){
transferStack();
}
}
/**
* 将入栈数据转入到出栈
*/
public void transferStack(){
while (!statckPush.empty()){
stackPop.push(statckPush.pop());
}
}
}