飞道的博客

剑指Offer对答如流系列 - 用两个栈实现队列

343人阅读  评论(0)

面试题8:用两个栈实现队列

一、题目描述

用两个栈实现一个队列。队列的声明如下,请实现它的两个函数appendTail和deleteHead,分别完成在队列尾部插入结点和在队列头部删除结点的功能。

二、思路

这道题非常简单,我给你说,你就把栈当成一个存放乒乓球(元素)的细瓶子,如果你有两个细瓶子,你就可以倒来倒去操纵乒乓球了


题中说用两个栈,换种角度,我们是拥有了两个存放乒乓球的细瓶子。

在队列尾部插入节点:
我们直接在栈1中插入即可,读取的时候,把栈1的元素移入栈2(细瓶子1的乒乓球倒入细瓶子2),栈2的遍历与队列保持一致即可。

队列头部删除结点的功能:
把栈1的元素移入栈2(细瓶子1的乒乓球倒入细瓶子2),移除栈2最顶端的的元素(拿走细瓶子2最上面的乒乓球)

三、实现

代码解答如下:

 class Queue{
        
        Stack<Integer> stack1 = new Stack<Integer>();
        Stack<Integer> stack2 = new Stack<Integer>();

        // 插入节点
        public void push(int node) {
            stack1.push(node);
        }

        // 删除节点
        public int pop() {
            if (stack2.empty()) {
                if (stack1.empty())
                    throw new RuntimeException("队列为空!");
                else {
                    while (!stack1.empty())
                        stack2.push(stack1.pop());
                }
            }
            return stack2.pop();
        }
    }

📚 快来,快来。打开CSDN APP扫一下下面二维码 或者点击 传送门 ,加入我的大本营(免费的),向我和其他人气博主提问哈。


转载:https://blog.csdn.net/qq_42322103/article/details/103994753
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场