[LeetCode] Implement Stack using Queues

Implement the following operations of a stack using queues.

• push(x) – Push element x onto stack.
• pop() – Removes the element on top of the stack.
• top() – Get the top element.
• empty() – Return whether the stack is empty.

Notes:

• You may assume that all operations are valid (for example, no pop or top operations will be called on an empty stack).
• Depending on your language, queue may not be supported natively. You may simulate a queue by using a list or deque (double-ended queue), as long as you use only standard operations of a queue – which means only push to back, pop from front, size, and is empty operations are valid.

解题思路

使用两个队列q1和q2，在q1上进行top()和push()操作，每次操作前把所有元素集中到q1中。

实现代码

// Runtime: 0 ms 暂时理解为计时故障
class Stack {
public:
    // Push element x onto stack.
    void push(int x) {
        gather(q1, q2);
        q1.push(x);
    }

    // Removes the element on top of the stack.
    void pop() {
        gather(q1, q2);
        while (q1.size() > 0 && q1.size() != 1)
        {
            q2.push(q1.front());
            q1.pop();
        }
        q1.pop();
    }

    // Get the top element.
    int top() {
        gather(q1, q2);
        while (q1.size() > 0 && q1.size() != 1)
        {
            q2.push(q1.front());
            q1.pop();
        }
        int n = q1.front();
        q2.push(q1.front());
        q1.pop();

        return n;
    }

    // Return whether the stack is empty.
    bool empty() {
        return q1.empty() && q2.empty();
    }
private:
    queue<int> q1;
    queue<int> q2;
    void gather(queue<int>& q1, queue<int>& q2)
    {
        while (!q2.empty())
        {
            q1.push(q2.front());
            q2.pop();
        }
    }
};

由于每次push()、top()或者pop()之后都有一个队列为空，因此直接将q1设为有元素的队列即可。可以省去出入队列的时间。

// Runtime: 0 ms
class Stack {
public:
    // Push element x onto stack.
    void push(int x) {
        toQ1(q1, q2);
        q1.push(x);
    }

    // Removes the element on top of the stack.
    void pop() {
        toQ1(q1, q2);
        while (q1.size() > 0 && q1.size() != 1)
        {
            q2.push(q1.front());
            q1.pop();
        }
        q1.pop();
    }

    // Get the top element.
    int top() {
        toQ1(q1, q2);
        while (q1.size() > 0 && q1.size() != 1)
        {
            q2.push(q1.front());
            q1.pop();
        }
        int n = q1.front();
        q2.push(q1.front());
        q1.pop();

        return n;
    }

    // Return whether the stack is empty.
    bool empty() {
        return q1.empty() && q2.empty();
    }
private:
    queue<int> q1;
    queue<int> q2;
    void toQ1(queue<int>& q1, queue<int>& q2)
    {
        if (q1.empty())
        {
            q1 = q2;
            queue<int> q;
            q2 = q;
        }
    }
};