LeetCode 225. 用队列实现栈(C++)

题目描述

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

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

注意:

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

解题思路

我们可以用两个队列(a,b)来模拟实现栈,其中会有一个临时的队列(b)负责在压人数据时帮助调换元素的顺序,使其满足栈后进先出的特点,再把调换过顺序的元素,放回到a队列,当出栈时,就直接从a队列中出已经调好顺序的元素就可以了。

以下是我写出的代码

class MyStack {
public:
    /** Initialize your data structure here. */
    MyStack() {
        
    }
    
    /** Push element x onto stack. */
    void push(int x) {
        queue  temp; //用来辅助调整元素顺序的临时队列
        temp.push(x);   //压入的元素首先压到临时队列中
        while (!a.empty()){ //把之前已将调整好的元素次序不变的a压入临时队列中,这时就把
                            //最后放入的元素压到了对列的第一个
            temp.push(a.front());
            a.pop();
        }
     
        while (!temp.empty()){ //这时再把调整好顺序的元素放到a队列中
            a.push(temp.front());
            temp.pop();
        }
        
    }
    
    /** Removes the element on top of the stack and returns that element. */
    int pop() {
        int temp=a.front();
        a.pop();
        return temp;
        
    }
    
    /** Get the top element. */
    int top() {
        return a.front();
        
    }
    
    /** Returns whether the stack is empty. */
    bool empty() {
        return a.empty();
        
    }
    private:
    queue   a;
};

/**
 * 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刷题之栈和队列)