LeetCode Implement Queue using Stacks 栈&队列

qwurey

于 2015-11-02 09:50:51 发布

阅读量325

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/yeruby/article/details/49582993

leetcode 专栏收录该内容

234 篇文章

订阅专栏

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路：

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

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();
    }
};