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

leetcode 剑指 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 <= values <= 10000
• 最多会对 appendTail、deleteHead 进行 10000 次调用

我的代码

class CQueue {
private:
    stack<int> tails, heads;
public:
    CQueue() {

    }

    void appendTail(int value) {
        tails.push(value);
    }

    int deleteHead() {
        if (tails.empty()) {
            return -1;
        }
        while (!tails.empty()) {
            heads.push(tails.top());
            tails.pop();
        }
        int res = heads.top();
        heads.pop();
        while (!heads.empty()) {
            tails.push(heads.top());
            heads.pop();
        }
        return res;
    }
};

我的成绩

执行结果：通过
执行用时：1120 ms, 在所有 C++ 提交中击败了24.09%的用户
内存消耗：111 MB, 在所有 C++ 提交中击败了100.00%的用户

一些想法

我写的比较简单，就是两个栈倒来倒去，应该有更好的方法。

执行用时为 0 ms 的范例

static int n=[](){std::ios::sync_with_stdio(false);std::cin.tie(nullptr);return 0;}();
class CQueue {
public:
    stack<int>a,b;
    CQueue() {

    }
    
    void appendTail(int value) {
        a.push(value);
    }
    
    int deleteHead() {
        if(b.empty()&&a.empty())
            return -1;
        else if(b.empty()&&!a.empty()){
            while(!a.empty()){
                b.push(a.top());
                a.pop();
            }
        }
        int temp=b.top();
        b.pop();
        return temp;
    }
};

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

思考

只有在栈b为空，栈a不为空时才需要进行翻转。。。