leetcode 232. Implement Queue using Stacks

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

原创最新推荐文章于 2023-01-14 11:29:42 发布 · 203 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

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

104 篇文章

订阅专栏

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

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