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

题目描述

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof

C++

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

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

题解

用第二个栈存第一个栈的元素实现第一个栈元素的倒置，然后删除第二个栈的栈顶，最后第一个栈存第二个栈。