【剑指Offer】用两个栈实现队列

题目描述

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

解题思路

模拟一遍：
（1）首先插入1，2，3，4，栈的情况：stack1={1,2,3,4}；stack2={}
（2）删除一个元素，由于1是最先插入的，所以1应该最先删除，因此需要将2，3，4压入stack2中，删除1，栈的情况：stack1={}；stack2={4,3,2}
（3）继续删除一个元素，2是第二个插入的，因此直接从stack2中出栈
（4）插入一个元素5，将5压入stack1
（5）继续删除两个元素，将stack2中3,4出栈
（6）再删除一个元素，stack2={}；stack1={5}，因此需要将5从stack1出栈，压入stac2，再出栈。

规律：

插入元素时，只向stack1压栈，删除元素时，首先看stack2是否有元素，有直接出栈返回，否则，看stack1是否有元素，有，则将stack1中所有元素出栈压入stack2，再出栈返回，否则，两个栈都为空。

代码

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.size() <= 0) {
            while(stack1.size() != 0) {
                Integer t = stack1.pop();
                stack2.push(t);
            }
        }
        return stack2.pop();
    }
}