LeetCode225 栈·用队列实现栈(C++)

题目描述：

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

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

注意:

你只能使用队列的基本操作-- 也就是 push to back, peek/pop from front, size, 和 is empty 这些操作是合法的。
你所使用的语言也许不支持队列。你可以使用 list 或者 deque（双端队列）来模拟一个队列 , 只要是标准的队列操作即可。
你可以假设所有操作都是有效的（例如, 对一个空的栈不会调用 pop 或者 top 操作）。

/*解题思路：利用双队列实现
创建两个队列 q1，q2
入栈：将元素 x 直接放入 q1 队列中。
出栈：也就是把 q1 的队尾元素出队列，由于队列只能从队头出队，因此先把 q1 中除了队尾元素的其他值存到 q2 中，再把队尾元素也就是栈顶出队，最后将 q2 中的值存到 q1 中。
获取栈顶元素：也就是获取 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();  
    }
};

/**
 * 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();
 */

LeetCode225 栈·用队列实现栈(C++)

猜你喜欢