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

}

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容