常考数据结构与算法:用两个栈实现队列

题目描述

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

分析
   队列的特性是：“先入先出”，栈的特性是：“先入后出”

当我们向模拟的队列插入数 1,2,3 时，假设插入的是 stack1，此时的栈情况为：

   栈 stack1：{1,2,3}

   栈 stack2：{}

当需要弹出一个数，根据队列的"先进先出"原则，1 先进入，则 1应该先弹出。但是此时 1在 stack1 的最下面，将 stack1 中全部元素逐个弹出压入 stack2，现在可以正确的从 stack2 中弹出 1，此时的栈情况为：

  栈 stack1：{}

  栈 stack2：{3,2}

继续弹出一个数，2比 3 先进入"队列"，2 弹出，注意此时 2在 stack2 的栈顶，可直接弹出，此时的栈情况为：

  栈 stack1：{}

  栈 stack2：{3}

此时向模拟队列插入一个数 4，还是插入 stack1，此时的栈情况为：

  栈 stack1：{4}

  栈 stack2：{3}

弹出一个数，3 比 4 先进入，3 弹出，注意此时 3 在 stack2 的栈顶，可直接弹出，此时的栈情况为：

  栈 stack1：{4}

  栈 stack2：{}

根据上述栗子可得出结论：

  当插入时，直接插入 stack1.

  当弹出时，当 stack2 不为空，弹出 stack2 栈顶元素，如果 stack2 为空，将 stack1 中的全部数逐个出栈入栈 stack2，再弹出 stack2 栈顶元素

import java.util.Stack;

public class StackQueue {

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