用栈实现队列操作【进阶:实现队列每个操作均摊时间复杂度为O(1)。】

用栈实现队列

需要完成队列的push,pop,peek取队列的第一个元素值,empty。

【不实现】每个操作均摊时间复杂度为O(1)的简单方法的思路如下:

需要用两个栈,当需要实现队列的push操作时,直接向s1压入元素,时间复杂度为O(1)。

要实现pop操作时,需要把s1的元素压入s2中,一直到s1只剩一个元素为止,并将该元素出栈不压入s2,最后将s2中的元素压回s1,时间复杂度为O(n)。

要实现peek操作(取栈顶元素),同样操作,只是不将s1的最后一个元素出栈而是返回它的值,最后还是将s2中的元素压回s1,时间复杂度为O(n)。

empty操作,只需要检查s1是否为空,时间复杂度为O(1)。

进阶:实现队列每个操作均摊时间复杂度为O(1)。

同样使用两个栈,s1专门负责push操作。

实现pop操作,当s2为空的时候,将s1中的元素全部压入栈中,然后直接对s2出栈,此时的出栈操作时间复杂度为O(1),也就是说只有当s2为空的时候执行pop操作才需要O(n)的时间复杂度。

peek操作同理。

empty操作判断s1和s2是否为空,都是O(1)的时间复杂度。

进阶的代码如下:

class MyQueue {
public:
    MyQueue() {

    }
    
    void push(int x) {
        s1.push(x);
    }
    
    int pop() {
        if(s2.empty())
        {  
           // s1不为空,s2为空,此时将s1的所有元素出栈并压入s2中,完成逆序
            s1Tos2();
        }
        int val = s2.top();
        s2.pop();
        return val;
    }
    
    int peek() {
        if(s2.empty())
        {  
            // s1不为空,s2为空,此时将s1的所有元素出栈并压入s2中,完成逆序
            s1Tos2();
        }
        return s2.top();
    }
    
    bool empty() {
        if(s2.empty() && s1.empty())
            return true;
        else
            return false;
    }

    void s1Tos2()  // s1不为空,s2为空,此时将s1的所有元素出栈并压入s2中,完成逆序
    {
        while(!s1.empty())
        {
            int temp = s1.top();
            s1.pop();
            s2.push(temp);
        }
    }

public:
    stack<int> s1;
    stack<int> s2;
};

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值