剑指offer 面试题9 用两个栈实现队列

1.题目

用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。

2.思路

两个栈s1,s2

push操作先将元素存在s1中

pop操作先判断s2是否为空

  • s2为空,s1的元素逐个弹出并压入s2
  • s2不为空,s2的栈顶元素是最先进入队列的元素,s2.pop()

3.实现

/*
    两个栈实现队列
    

*/
class MyQueue {
public:
    MyQueue() {
        
    }
    void push(int x) {
        stIn.push(x);
    }
    
    int pop() {
        // 不用重复 while 循环,直接调用 peek() 就可以
        int popNum = peek();
        stOut.pop();
        return popNum;
    }
    
    int peek() {
        // stOut 空的情况下才会指向这样的操作
        if (stOut.empty()) {
            while (!stIn.empty()) {
                stOut.push(stIn.top());
                stIn.pop();
            }
        }
        return stOut.top();
    }
    
    bool empty() {
        return stIn.empty() && stOut.empty();
    }
private:
    stack<int> stIn;
    stack<int> stOut;
};

/**
 * 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();
 */

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值