剑指offer_4_两个栈实现队列

两个栈实现队列

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

思路：

有两个栈，一个专门负责进入，另一个专门负责输出，当遇到pop时，如果负责输出,不为空，直接pop,如果负责输出的为空，则将负责进入的出栈，并压入负责输出的栈，然后直接pop,

static class queue{
       Stack<Integer> stack1 = new Stack<Integer>();
       Stack<Integer> stack2 = new Stack<Integer>();


       public void push(int node) {
           stack1.add(node);
       }

       public int pop() {
           if(!stack2.isEmpty()){
               return stack2.pop();
           }
           else {
               while(!stack1.isEmpty()){
                   stack2.add(stack1.pop());
               }
               return stack2.pop();
           }
       }
   }