面试题9：用两个栈实现队列

import java.util.Stack;

/**

 * 面试题9：用两个栈实现队列

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

思路：一个栈压入元素，而另一个栈作为缓冲，将栈1的元素出栈后压入栈2中。也可以将栈1中的最后一个元素直接出栈，而不用压入栈2中再出栈。

 * @author juaner

 *

 */

public class TwoStacksQueue {

 

    public static void main(String[] args) {

       TwoStacksQueue tsq = new TwoStacksQueue();

       tsq.push(5);

       tsq.push(6);

        tsq.push(7);

       System.out.println(tsq.peek());

       tsq.push(8);

       tsq.push(9);

       tsq.pop();

       tsq.pop();

       tsq.pop();

       System.out.println(tsq.peek());

    }

 

     Stack<Integer> stack1 = new Stack<Integer>();

        Stack<Integer> stack2 = new Stack<Integer>();

      

        public void push(int node) {

            stack1.push(node);

        }

       

        public int pop() {

            if (stack1.isEmpty() && stack2.isEmpty()) {

            throw new RuntimeException("Queue is empty!");

            }

           

            if(stack2.isEmpty()){

                while (!stack1.isEmpty()) {

                    stack2.push(stack1.pop());

                }

            }

            return stack2.pop();

        }

       

        public int peek() {

            if (stack1.isEmpty() && stack2.isEmpty()) {

            throw new RuntimeException("Queue is empty!");

            }

           

            if(stack2.isEmpty()){

                while (!stack1.isEmpty()) {

                    stack2.push(stack1.pop());

                }

            }

            return stack2.peek();

        }

}