对“积压订单中的订单总数“算法的优化

二话不说,看下优化后的效果如下:

  • 时间从356ms缩减为188ms,内存消耗也有减少
  • 可以看出map<int,int> mp所涉及的删除操作是没有优先队列的时间复杂度优秀的

 

 题目描述如下:

给你一个二维整数数组 orders ,其中每个 orders[i] = [pricei, amounti, orderTypei] 表示有 amounti 笔类型为 orderTypei 、价格为 pricei 的订单。

订单类型 orderTypei 可以分为两种:

0 表示这是一批采购订单 buy
1 表示这是一批销售订单 sell

注意,orders[i] 表示一批共计 amounti 笔的独立订单,这些订单的价格和类型相同。对于所有有效的 i ,由 orders[i] 表示的所有订单提交时间均早于 orders[i+1] 表示的所有订单。

存在由未执行订单组成的 积压订单 。积压订单最初是空的。提交订单时,会发生以下情况:

如果该订单是一笔采购订单 buy ,则可以查看积压订单中价格 最低 的销售订单 sell 。如果该销售订单 sell 的价格 低于或等于 当前采购订单 buy 的价格,则匹配并执行这两笔订单,并将销售订单 sell 从积压订单中删除。否则,采购订单 buy 将会添加到积压订单中。
反之亦然,如果该订单是一笔销售订单 sell ,则可以查看积压订单中价格 最高 的采购订单 buy 。如果该采购订单 buy 的价格 高于或等于 当前销售订单 sell 的价格,则匹配并执行这两笔订单,并将采购订单 buy 从积压订单中删除。否则,销售订单 sell 将会添加到积压订单中。
输入所有订单后,返回积压订单中的 订单总数 。由于数字可能很大,所以需要返回对 10^9 + 7 取余的结果。

 代码如下:

struct buy{
    int price;
    int amount;
    bool operator <(const buy &t) const{
        return price<t.price;
    }
};
struct sell{
    int price;
    int amount;
    bool operator <(const sell &t) const{
        return price>t.price;
    }
};
class Solution {
public:
    int getNumberOfBacklogOrders(vector<vector<int>>& orders) {
        priority_queue<buy>b;
        priority_queue<sell>s;
        long long num=0;
        for(auto &t:orders)
            if(t[2]==1){
                if(b.empty()){
                    s.push(sell{t[0],t[1]});
                    continue;
                }
                auto bu=b.top();
                while(!b.empty()&&t[1]!=0){  //要么订单处理结束,要么已经用完所有的可用buy订单
                    if(bu.price<t[0]) break;
                    if(bu.amount>t[1]){
                        bu.amount-=t[1];
                        t[1]=0;
                        b.pop();
                        b.push(bu);
                    }
                    else{
                        t[1]-=bu.amount;
                        b.pop();
                        bu=b.top();
                    }
                }
                if(t[1]>0)
                    s.push(sell{t[0],t[1]});
            }
            else{
                if(s.empty()){
                    b.push(buy{t[0],t[1]});
                    continue;
                }
                auto se=s.top();
                while(!s.empty()&&t[1]!=0){
                    if(se.price>t[0]) break;
                    if(se.amount>t[1]){
                        se.amount-=t[1];
                        t[1]=0;
                        s.pop();
                        s.push(se);
                    }
                    else{
                        t[1]-=se.amount;
                        s.pop();
                        se=s.top();
                    }
                }
                if(t[1]>0)
                    b.push(buy{t[0],t[1]});
            }
        while(!s.empty()){
            auto t=s.top();
            num+=t.amount;
            s.pop();
            if(num>(int)pow(10,9)+7)
                num=num%((int)pow(10,9)+7);
        }
        while(!b.empty()){
            auto t=b.top();
            num+=t.amount;
            b.pop();
            if(num>(int)pow(10,9)+7)
                num=num%((int)pow(10,9)+7);
        }
        return num;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值