剑指offer 面试题9 用两个栈实现队列

1.题目

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

2.思路

两个栈s1，s2

操作先将元素存在s1中

操作先判断s2是否为空

• s2为空，s1的元素逐个弹出并压入s2
• s2不为空，s2的栈顶元素是最先进入队列的元素，s2.pop()

3.实现

/*
    两个栈实现队列
    

*/
class MyQueue {
public:
    MyQueue() {
        
    }
    void push(int x) {
        stIn.push(x);
    }
    
    int pop() {
        // 不用重复 while 循环，直接调用 peek() 就可以
        int popNum = peek();
        stOut.pop();
        return popNum;
    }
    
    int peek() {
        // stOut 空的情况下才会指向这样的操作
        if (stOut.empty()) {
            while (!stIn.empty()) {
                stOut.push(stIn.top());
                stIn.pop();
            }
        }
        return stOut.top();
    }
    
    bool empty() {
        return stIn.empty() && stOut.empty();
    }
private:
    stack<int> stIn;
    stack<int> stOut;
};

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