Java 使用栈实现简单队列功能

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());
        }
    }

}