LeetCode 225. 用队列实现栈

题目描述

使用队列实现栈的下列操作：

• push(x) -- 元素 x 入栈
• pop() -- 移除栈顶元素
• top() -- 获取栈顶元素
• empty() -- 返回栈是否为空

思路

创建两个队列 q1，q2

1. 入栈

   将元素 x 直接放入 q1 队列中。

2. 出栈

   • 也就是把 q1 的队尾元素出队列，由于队列只能从队头出队，因此先把 q1 中除了队尾元素的其他值存到 q2 中
   • 再把队尾元素也就是栈顶出队
   • 最后将 q2 中的值存到 q1 中

3. 获取栈顶元素

   也就是获取 q1 的队尾元素

class MyStack {
public:
    queue<int> q1, q2;
    int i = 0;
    int temp;
    int res;
    MyStack() {
        
    }

    void push(int x) {
        q1.push(x);
    }
    
    /** Removes the element on top of the stack and returns that element. */
    int pop() {
        int length1 = q1.size();
        i = 0;
        while(length1 > 1 && i < length1 - 1)
        {
            temp = q1.front();
            q2.push(temp);
            q1.pop();
            i++;
        }
        res = q1.front();
        q1.pop();
        i = 0;
        int length2 = q2.size();
        while(i < length2)
        {
            temp = q2.front();
            q1.push(temp);
            q2.pop();
            i++;
        }
        return res;
    }
    
    /** Get the top element. */
    int top() {
        return q1.back();
    }
    
    /** Returns whether the stack is empty. */
    bool empty() {
        return q1.empty();  
    }
};