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

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

题目描述：用两个栈实现一个队列。队列的声明如下，请实现它的两个函数 appendTail 和 deleteHead ，分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素，deleteHead 操作返回 -1 )
示例 1：
输入：
[“CQueue”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[3],[],[]]
输出：[null,null,3,-1]
示例 2：
输入：
[“CQueue”,“deleteHead”,“appendTail”,“appendTail”,“deleteHead”,“deleteHead”]
[[],[],[5],[2],[],[]]
输出：[null,-1,null,null,5,2]

思路

要用两个栈实现队列，那么需要满足的是将栈的先入后出转化为先入先出。可以想象若输入的为1 2 3 4 5，那么最后输入的5将在第一个栈的栈顶，若要实现队列，在队列尾部插入元素，只需要在第一个栈中入栈即可，而删除头部元素的话，头部元素在第一个栈的栈底，因此删除则必须将第一个栈的元素依次出栈放入第二个栈，此时第二个栈的出栈顺序就是队列的出队顺序。但是需要考虑到第一个栈与第二个栈都为空的情况，此时则说明队列中没有元素，返回-1。同时要考虑二号栈是否为空，为空时才能够将第一个栈的元素依次移入第二个栈。

class CQueue {
    stack<int> stack1,stack2;
public:
    CQueue() {
        
    }
    
    void appendTail(int value) {
        stack1.push(value);
    }
    
    int deleteHead() {
        int ans;
        if(stack2.empty()){
            while(!stack1.empty()){
                stack2.push(stack1.top());
                stack1.pop();
            }
        }
        if(stack2.empty()){
            return -1;
        }else{
            ans = stack2.top();
            stack2.pop();
            return ans;
        }

    }
};

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