Cracking the coding interview--Q3.5

题目

原文：

Implement a MyQueue class which implements a queue using two stacks.

译文：

用两个栈实现一个队列MyQueue。

解答

堆栈和队列的最大区别在于：一个是先进后出，一个是后进先出。用两个栈来实现队列的最简单方式是：进入队列则往第一个栈压栈， 出队列则将第一个栈的数据依次压入第二个栈，然后出栈。先所有的元素进堆栈S1，这样最先进栈的元素在栈底，最后的在栈顶；然后将S1中的元素依次弹出，并压入堆栈S2中。这样S2中先进的元素就在栈顶后进的就在栈底了。
但是如果对队列和出队列的动作反复进行的话，我就要在S1和S2两个堆栈中反复的倒来倒去。其实有偷懒的方法。
当有元素要进入队列的时候，直接压如堆栈S1中，元素需要从队列中出列的话，检查S2中是否有元素，如果没有再采用之前的方法将S1中的元素“倒”入到S2中，如果S2中非空在直接弹出元素即为队列中出列的元素。这样的方法就免于在两个堆栈之间倒来倒去了.

代码如下：

public class MyQueue<T>
{
    Stack<T> s1, s2;
    public MyQueue()
    {
        s1 = new Stack<T>();
        s2 = new Stack<T>();
    }

    public int size()
    {
        return s1.size() + s2.size();
    }

    public void add(T value)
    {
        s1.push(value);
    }

    public T peek()
    {
        if (!s2.empty()) return s2.peek();
        while (!s1.empty()) s2.push(s1.pop());
        return s2.peek();
    }

    public T remove()
    {
        if (!s2.empty()) return s2.pop();
        while (!s1.empty()) s2.push(s1.pop());
        return s2.pop();
    }
}

---EOF--