LeetCode Implement Queue using Stacks 栈&队列

最新推荐文章于 2022-08-14 12:35:34 发布

原创最新推荐文章于 2022-08-14 12:35:34 发布 · 338 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

234 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列基本操作的方法。通过不断转移栈内元素的位置，可以实现在队列尾部添加元素(push)、从队列头部移除元素(pop)、获取队列头部元素(peek)及判断队列是否为空(empty)等功能。

思路：

用两个栈实现队列的功能。

c++ code：

class Queue {
public:
    stack<int> s, tmp;
    // Push element x to the back of queue.
    void push(int x) {
        s.push(x);
    }

    // Removes the element from in front of queue.
    void pop(void) {
        while(s.size() > 1) {
            tmp.push(s.top());
            s.pop();
        }
        s.pop();
        while(tmp.size() > 0) {
            s.push(tmp.top());
            tmp.pop();
        }
    }

    // Get the front element.
    int peek(void) {
        while(s.size() > 1) {
            tmp.push(s.top());
            s.pop();
        }
        int v = s.top();
        while(tmp.size() > 0) {
            s.push(tmp.top());
            tmp.pop();
        }
        return v;
    }

    // Return whether the queue is empty.
    bool empty(void) {
        return s.empty();
    }
};