232. Implement Queue using Stacks

仅供自己学习

 

思路：

这里只用两个栈来模拟队列，因为栈是先进后出，队列是先进先出，所以一个栈instack用来存进来的元素，另一个栈outstack存instack弹出的元素，且若instack要将元素加入进outstack中，只有当outstack空了才能重新加入新进instack得元素，否则顺序将错误。这样队列就能模拟成功

对于每一种功能：

push，只需要将新元素加入instack中即可，暂不需要加入outstack，我们会在pop中加入

pop，首先判断 outstack有没有空，如果有就把instack的元素加入进outstack然后再获取outstack的第一个元素，再pop出来，否则直接top，pop即可

peek，依然是判断outstack有没有空，如果空了就从instack将元素加入，然后return top即可 不用pop

empty，直接判断两个栈有没有空，没有就false，有就true

代码：

class MyQueue {
private:
    stack<int> inst,outst;
    void intoout(){
        while(!inst.empty()){
            outst.push(inst.top());
            inst.pop();
        }
    }
public:
    /** Initialize your data structure here. */
    MyQueue() {

    }
    
    /** Push element x to the back of queue. */
    void push(int x) {
        inst.push(x);
    }
    
    /** Removes the element from in front of queue and returns that element. */
    int pop() {
        if(outst.empty()){
            intoout();
        }
        auto temp=outst.top();
        outst.pop();
        return temp;
    }
    
    /** Get the front element. */
    int peek() {
        if(outst.empty())
            intoout();
        return outst.top();
    }
    
    /** Returns whether the queue is empty. */
    bool empty() {
        if(inst.empty()&&outst.empty())
            return true;
            return false;
    }
};

/**
 * Your MyQueue object will be instantiated and called as such:
 * MyQueue* obj = new MyQueue();
 * obj->push(x);
 * int param_2 = obj->pop();
 * int param_3 = obj->peek();
 * bool param_4 = obj->empty();
 */