[剑指OFFER]之栈和队列：JZ9 用两个栈实现队列

1.题目

用两个栈来实现一个队列，使用n个元素来完成 n 次在队列尾部插入整数(push)和n次在队列头部删除整数(pop)的功能。 队列中的元素为int类型。保证操作合法，即保证pop操作时队列内已有元素。

2.解法

2.1 提供的基础结构

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
    }
    
    public int pop() {
    }
}

2.2 解法1

栈实现队列要通过2个栈，第一个栈负责压入，第二个栈负责弹出。

如果第二个栈是空的 没有数据了，就再把第二个栈全部压入过来，记住，是全部数据都倒过来。

在倒过来之后，数据的顺序就正常了，就是先进先出的类型了。

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() {
        if (!stack2.isEmpty()) return stack2.pop();
        while (!stack1.isEmpty())
            stack2.push(stack1.pop());
        return stack2.pop();
    }
}

总结

栈实现队列要通过2个栈，第一个栈负责压入，第二个栈负责弹出。

算法系列在github上有一个开源项目，主要是本系列博客的demo代码。https://github.com/forestnlp/alg
如果您对软件开发、机器学习、深度学习有兴趣请关注本博客，将持续推出Java、软件架构、深度学习相关专栏。
您的支持是对我最大的鼓励。