剑指Offer系列—— 09. 用两个栈实现队列

题目

用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )

示例

输入：
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]
输出：[null,null,3,-1]

思路

我们第一个栈就正常的放元素。第二个栈，一般情况的话，是将第一个栈中的所有元素按序依次放入，之后栈顶元素就是所谓的队头元素了；再比如，进行多次操作后，栈2中还有元素，那么其栈顶就是我们要找的；再比如，队列中无元素，我们需要返回-1。代码如下

代码

class CQueue {
public:
    stack<int> stk1, stk2;
    CQueue() {

    }
    
    void appendTail(int value) {
        stk1.push(value);
    }
    
    int deleteHead() {
        if (!stk2.empty()) {
            int t = stk2.top();
            stk2.pop();
            return t;
        }
        while (!stk1.empty()) {
            stk2.push(stk1.top());
            stk1.pop();
        }
        if (stk2.empty()) return -1;
        int t = stk2.top();
        stk2.pop();
        return t;
    }
};

/**
 * Your CQueue object will be instantiated and called as such:
 * CQueue* obj = new CQueue();
 * obj->appendTail(value);
 * int param_2 = obj->deleteHead();
 */