【Leetcode-算法】232. 用栈实现队列(C++)

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列的支持的所有操作(push、pop、peek、empty):

实现 MyQueue 类:

void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空,返回 true ;否则,返回 false

思路:

        其实就是两个栈倒来倒去~适当的时间倒可以提高效率。

        我们只要保证,要输出的那个元素是最早的输入就行了,所以不需要每一次输入输出都倒,只要out栈为空的时候倒一次in即可。

AC代码:

class MyQueue {
private:
    stack<int> in,out;
    void in2out(){
        while(!in.empty()){
            out.push(in.top());
            in.pop();
        }
    }

public:
    MyQueue() {}

    void push(int x) {
        in.push(x);
    }
    
    int pop() {
        if(out.empty()) in2out();
        int temp = out.top();
        out.pop();
        return temp;
    }
    
    int peek() {
        if(out.empty()) in2out();
        return out.top();
    }
    
    bool empty() {
        if(in.empty()&&out.empty()) return true;
        else return false;
    }
};

        最后bool empty()的写法还可以优美一点~

        return in.empty()&&out.empty();

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值