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

题目描述

题目链接

思路

代码1

class CQueue {
    
public:
//加入队尾 appendTail() ： 将数字 val 加入栈 A 即可。
// 删除队首deleteHead() ： 有以下三种情况。
// 当栈 B 不为空： B中仍有已完成倒序的元素，因此直接返回 B 的栈顶元素。
// 否则，当 A 为空： 即两个栈都为空，无元素，因此返回 -1 。
// 否则： 将栈 A 元素全部转移至栈 B 中，实现元素倒序，并返回栈 B 的栈顶元素。
    stack<int> A,B;
    CQueue() {
     }
    void appendTail(int value) {
    
        A.push(value);
    }
    
    int deleteHead() {
    
        if(!B.empty()) {
    
            int tmp = B.top();
            B.pop();
            return tmp;
        }
        if(A.empty()) {
    
            return -1;
        }
          while(!A.empty()) {
    
            int tmp = A.top();
            A.pop();
            B.push(tmp);
            }
        int tmp = B.top();
        B.pop();    
        return tmp;
    }
};

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

感谢阅读