两个栈实现队列

题目

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

分析

栈的特点是先进后出，而队列是先进先出。我们有两个栈s1, s2, 可以s1专门用于接收push，pop的时候，将s1 中的元素 全部pop到s2 中，这样s2的栈顶就是s1 的栈底，也就是最先入栈的元素，从而整体上就是先进先出。

代码实现

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