leetcode 232: Implement Queue using Stacks

最新推荐文章于 2021-05-13 14:24:56 发布

原创最新推荐文章于 2021-05-13 14:24:56 发布 · 222 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

201 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法。通过在需要弹出元素时将第一个栈中的所有元素（除了最后一个）转移到第二个栈中，然后弹出第一个栈的最后一个元素。之后再将第二个栈中的元素转回第一个栈，以确保队列操作的正确执行。

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

Use two stacks. Every time I need to pop, I push all numbers in st1 into st2 except the last one, then I pop that one. After that, I push back all numbers into st1. Don't forget to update the front variable.

class Queue {
public:
    // Push element x to the back of queue.
    void push(int x) {
        if(st1.empty())
            front=x;
        st1.push(x);
    }

    // Removes the element from in front of queue.
    void pop(void) {
        while(st1.size()!=1)
        {
            st2.push(st1.top());
            st1.pop();
        }
        st1.pop();
        if(!st2.empty())
            front=st2.top();
        while(!st2.empty())
        {
            st1.push(st2.top());
            st2.pop();
        }
    }

    // Get the front element.
    int peek(void) {
        return front;
    }

    // Return whether the queue is empty.
    bool empty(void) {
        return st1.empty();
    }
private:
    stack<int> st1;
    stack<int> st2;
    int front;
};