剑指Offer——用两个栈实现队列

题目描述

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

时间限制：1秒 空间限制：32768K 热度指数：383776

本题知识点： 队列 栈

代码实现

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
        stack1.push(node);
    }
    
    public int pop() {
        while(!stack1.isEmpty()){
            stack2.push(stack1.pop());
        }
        int result = stack2.pop();
        while(!stack2.isEmpty()){
            stack1.push(stack2.pop());
        }
        return result;
    }
}

解题思路

做这道题之前我们首先需要了解队列和栈的相关知识，大家可以看这两篇文章数据结构与算法系列——栈和数据结构与算法系列——队列。我们知道了栈和队列的特点之后，这道题就简单多了，我们用其中一个栈 stack1 来存放数据，每当有新的数据入队的时候我们就执行入栈操作；另一个栈 stack2 只是在出队的时候充当一个临时存放数据的作用，因为队列的特点是先进先出，而栈的特点时先进后出，所以当执行出队操作时，我们首先把 stack1 中的所有数据转存到 stack2 中，这样最先入队的数据就到了 stack2 栈顶的位置，而最后入队的数据就到了栈底的位置，然后我们对 stack2 执行出栈操作，完成队列的出队操作。完成之后我们再把 stack2 中剩余的数据放回到 stack1 中。

这个过程描述起来有点啰嗦，我们通过画图直观的看一下，相信看完你马上就明白了。

欢迎关注公众号：「努力给自己看」