【LintCode】用栈实现队列

描述

正如标题所述，你需要使用两个栈来实现队列的一些操作。

队列应支持push(element)，pop() 和 top()，其中pop是弹出队列中的第一个(最前面的)元素。

pop和top方法都应该返回第一个元素的值。

样例

比如push(1), pop(), push(2), push(3), top(), pop()，你应该返回1，2和2

思路

    s1是入栈的，s2是出栈的。

• 入队列：直接压入s1即可
• 出队列：如果s2不为空，把s2中的栈顶元素直接弹出；否则，把s1的所有元素全部弹出压入s2中，再弹出s2的栈顶元素

代码

<span style="font-size:18px;">public class Queue {
    private Stack<Integer> stack1;
    private Stack<Integer> stack2;


    public Queue() {
       // do initialization if necessary
       stack1=new Stack<Integer>();
       stack2=new Stack<Integer>();
    }
    
    public void push(int element) {
        // write your code here
        stack1.push(element);
    }


    public int pop() {
        // write your code here
       if(stack2.empty())
       while(!stack1.empty()){
           int s1=stack1.pop();
           stack2.push(s1);
       }
  
       
      return stack2.pop();
    }


    public int top() {
        // write your code here
     if(stack2.empty())
     while(!stack1.empty()){
       int s1=stack1.pop();
       stack2.push(s1);
     }
     int s2=stack2.peek();//这里只是返回第一个元素的值，并没有弹出第一个元素
     //Java中返回堆栈顶的值用peek ,C++用top
    
     return s2;
        
    }
}</span>