剑指Offer对答如流系列

面试题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();
        }
    }
原文地址:https://www.cnblogs.com/JefferyChenXiao/p/12246271.html