225. 用队列实现栈

发布时间 2023-03-24 19:20:26作者: xiazichengxi

请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。

实现 MyStack 类:

void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。

class MyStack {
public:
    MyStack() = default;
    void get_first(std::queue<int> &q1)
    {
        int len = q1.size();
        if(len>1)
        {
            for(int i = 0; i < len-1; i++ )
            {
                int tmp =  q1.front();
                q1.pop();
                q1.push(tmp);
            }
        }
    }
    void push(int x) {
        if(q2.empty())
        {
            q2.push(x);
        }
        else{
            q2.push(x);
            int tmp = q2.front();
            q2.pop();
            q1.push(tmp);
            get_first(q1);
        }
    }
    
    int pop() {
        int res = 0;
        if(!q2.empty())
        {
            res = q2.front();
            q2.pop();
        }
        else if(!q1.empty())
        {
            res = q1.front();
            q1.pop();
        }
        return res;
    }
    
    int top() {
        int res = 0;
        if(!q2.empty())
        {
            res = q2.front();
        }
        else if(!q1.empty())
        {
            res = q1.front();
        }
        return res;
    }
    
    bool empty() {
        if(!q1.empty()||!q2.empty()) return false;
        else return true;
    }
private:
    std::queue<int> q1;
    std::queue<int> q2;
};

/**
 * Your MyStack object will be instantiated and called as such:
 * MyStack* obj = new MyStack();
 * obj->push(x);
 * int param_2 = obj->pop();
 * int param_3 = obj->top();
 * bool param_4 = obj->empty();
 */