ACwing刷题笔记 20.用两个栈实现队列

ACwing刷题笔记 20.用两个栈实现队列

代码

思路：建立一个输入栈和一个输出栈，将所有的元素先进入输入栈，每次要输出元素时，先将输入栈中的元素全部复制到输出栈中，然后输出栈pop出一个元素，之后将输出栈元素复制到输入栈中（取栈顶元素就不需要pop）

class MyQueue {
public:
    /** Initialize your data structure here. */
    stack<int> stk,cache;
    MyQueue() {
        
    }
    
    
    // 输入栈 和输出栈
    // 每次要输出元素 都需要把输入栈中的元素全部复制到输出栈中 然后pop 之后复制回去
    
    
    /** Push element x to the back of queue. */
    void push(int x) {
        stk.push(x);// 输入元素
    }
    
    void copy(stack<int> &a,stack<int> &b)
    {
        //输入栈元素 全部 复制到输出栈中
        while(a.size()){
            b.push(a.top());
            a.pop();
        }
    }
    
    /** Removes the element from in front of queue and returns that element. */
    int pop() {
        // 出队列 将输入栈中的元素全部压入输出栈
        copy(stk,cache);
        int res = cache.top();// 取出栈顶元素
        cache.pop();// 输出栈顶元素
        copy(cache,stk);// 输出完之后  复制回去
        return res;
    }
    
    /** Get the front element. */
    int peek() {
        copy(stk,cache);
        int res = cache.top();
        copy(cache,stk);
        return res;
    }
    
    /** Returns whether the queue is empty. */
    bool empty() {
        return stk.empty();// 判断输入栈是否有元素
        
    }
};

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