225. 用队列实现栈

    xiaoxiao2022-07-07  216

    题目描述: 

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

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

    注意:

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

    算法1:

    使用两个队列实现

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

    算法2:

    使用一个队列实现

    class MyStack { queue<int> nums; public: /** Initialize your data structure here. */ MyStack() { //nothing to do } /** Push element x onto stack. */ void push(int x) { nums.push(x); //将前面的size-1个元素放到后面去 for(int i = 0; i < nums.size() - 1; i++){ nums.push(nums.front()); nums.pop(); } } /** Removes the element on top of the stack and returns that element. */ int pop() { int num = nums.front(); nums.pop(); return num; } /** Get the top element. */ int top() { return nums.front(); } /** Returns whether the stack is empty. */ bool empty() { return nums.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(); */

     

    最新回复(0)