zl程序教程

您现在的位置是:首页 >  后端

当前栏目

81、【栈与队列】leetcode ——232. 用栈实现队列(C++版本)

C++LeetCode队列队列 实现 版本 81 232
2023-09-11 14:20:01 时间

题目描述

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

原题链接:232. 用栈实现队列

使用两个对顶栈实现

image.png
因为只能用栈的后进先出的方式实现队列先进先出的方式,因此就使用两个栈来实现。

一个stack用于输入元素,另一个stack用于输出元素。

每次先输入的会被压入到stack-in的栈底,当想要实现队列的方式输出时,需要让stack-in的栈底元素输出。因此就再用另一个stack-out作为媒介,将stack-in中元素全部压入stack-out中,此时stack-in中的栈底元素,就变成了stack-out中的栈顶元素。此时输出stack-out中元素时,先被输出的便是stack-in中的栈底元素。从而实现了队里的FIFO操作方式。

class MyQueue {
public:
    stack<int> sIn;
    stack<int> sOut;
    MyQueue() {
        
    }
    
    void push(int x) {
        sIn.push(x);        
    }
    
    int pop() {        
        // 当sOut里为空时,将sIn中元素都加入压入进去
        if(sOut.empty()) {
            while(!sIn.empty()) {
                sOut.push(sIn.top());
                sIn.pop();
            }
        }
        // 当sOut里不为空时,获取sOut的栈顶元素弹出
        int res = sOut.top();
        sOut.pop();
        return res;
    }
    
    int peek() {
        // 复用已定义过的pop()方法
        int res = this->pop();
        // 因为为获取队头元素,而不弹出,因此需要压回栈中
        sOut.push(res);
        return res;
    }
    
    bool empty() {
        return sOut.empty() && sIn.empty();
    }
};

/**
 * Your MyQueue object will be instantiated and called as such:
 * MyQueue* obj = new MyQueue();
 * obj->push(x);
 * int param_2 = obj->pop();
 * int param_3 = obj->peek();
 * bool param_4 = obj->empty();
 */

image.png

参考文章:232.用栈实现队列用栈实现队列