剑指offer-用两个栈实现队列（leetcode）

最新推荐文章于 2022-04-11 12:30:27 发布

原创最新推荐文章于 2022-04-11 12:30:27 发布 · 140 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #剑指offer

算法学习专栏收录该内容

30 篇文章

订阅专栏

本文介绍了一种使用两个栈来模拟队列操作的方法。通过将数据先压入栈s1，再转移到栈s2，可以实现先进先出的队列特性。在出栈时，先检查s2是否为空，如果为空则将s1的所有元素倒入s2，最后返回s2的栈顶元素并移除。

这道题目的思路很简单：将输入的数据压入s1中，再将s1中的数据压入s2中，这样s2的出栈顺序就和队列的出队顺序一致。

但是在判断出栈的时候要先判断s2是否为空，再判断s1是否为空。

若两者都为空，则输出-1

若s2不为空，则返回栈顶元素

若s2为空 && s1不为空，将s1中的元素压入s2中，再返回s2的栈顶元素

class CQueue {
    stack<int> s1;
    stack<int> s2;
public:
    CQueue() {

    }
    
    void appendTail(int value) {
        s1.push(value);
    }
    
    int deleteHead() {
        int ans = -1;
        if(!s2.empty()){
            ans = s2.top();
            s2.pop();
            return ans;
        }
        if(s1.empty()){
            return ans;
        }

        while(!s1.empty()){
            s2.push(s1.top());
            s1.pop();
        }
        ans = s2.top();
        s2.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();
 */