leetcode 232. Implement Queue using Stacks

最新推荐文章于 2023-01-14 11:29:42 发布

GDRetop

最新推荐文章于 2023-01-14 11:29:42 发布

阅读量194

点赞数

CC 4.0 BY-SA版权

分类专栏： ##笔试类编程题文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/firenet1/article/details/52263749

##笔试类编程题专栏收录该内容

104 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法。通过定义一个类Queue，该类包括push、pop、peek和empty等方法，实现了队列的基本操作。push方法用于将元素添加到队列尾部；pop方法移除队头元素；peek方法返回队头元素但不移除；empty方法判断队列是否为空。

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

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

    // Removes the element from in front of queue.
    void pop(void) {
        stack<int>res;
        while(ans.size() > 1){
            res.push(ans.top());
            ans.pop();
        }
        if(ans.size() > 0) ans.pop();
        while(res.size() > 0){
            ans.push(res.top());
            res.pop();
        }
    }

    // Get the front element.
    int peek(void) {
        
        if(ans.size() == 0) return 0;
        stack<int> res;
        while(ans.size() > 0){
            res.push(ans.top());
            ans.pop();
        }
        int x =  res.top();
        while(res.size() > 0){
            ans.push(res.top());
            res.pop();
        }
        return x;
    }

    // Return whether the queue is empty.
    bool empty(void) {
        return ans.size() == 0;
    }
};