leetcode 225.用队列实现栈

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

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

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/implement-stack-using-queues
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:
类似于用两个栈实现队列,两个队列同样也可以用于来实现栈。主要是在push新元素的时候,push到两个队列中为空的队列,之后将另一个队列中的所有元素push过来即可。为了节省时间,可以设置flag位,使得每一次的push可以避免再一次的返回到固定队列的过程。

class MyStack {
public:
    /** Initialize your data structure here. */
    queue<int> q1;
    queue<int> q2;
    int flag = 1;
    MyStack() {
        
    }
    
    /** Push element x onto stack. */
    void push(int x) {
        if(flag==1)
        {
            q2.push(x);
            while(!q1.empty())
            {
                q2.push(q1.front());
                q1.pop();
            }
            flag = 2;
            return;
        }
        if(flag==2)
        {
            q1.push(x);
            while(!q2.empty())
            {
                q1.push(q2.front());
                q2.pop();
            }
            flag = 1;
            return;
        }
    }
    
    /** Removes the element on top of the stack and returns that element. */
    int pop() {
        if(flag==1)
        {
            int temp = q1.front();
            q1.pop();
            return temp;
        }
        if(flag==2)
        {
            int temp = q2.front();
            q2.pop();
            return temp;
        }
    return -1;
    }
    
    /** Get the top element. */
    int top() {
        if(flag==1)return q1.front();
        if(flag==2)return q2.front();
        return -1;
    }
    
    /** Returns whether the stack is empty. */
    bool empty() {
        return !q1.empty()^q2.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();
 */

你可能感兴趣的:(leetcode)