积压订单数量(233周赛)()

给你一个二维整数数组 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 将会添加到积压订单中。
输入所有订单后,返回积压订单中的 订单总数 。由于数字可能很大,所以需要返回对 109 + 7 取余的结果。

 1.Buy买入价格放入大顶堆,每次返回Buy的最大价格;
2.Sell卖出价格放入小顶堆,每次返回Sell的最小价格;
3.大小顶堆的元素结构为pair<int, int>; //<价格, 数目>
4.若当前order[2]为0,即当前订单是Buy,需要找最小的Sell,直接返回SellQ最后一个元素即可;
5.如当前order[2]为1,即当前订单是Sell,需要找最大的Buy,直接返回BuyQ最后一个元素即可;
6.最后统计SellQ和BuyQ中元素的总数目;

class Solution {
public:
    int getNumberOfBacklogOrders(vector<vector<int>>& orders) {
        priority_queue<pair<int, int>, vector<pair<int, int>>, less<pair<int, int>>> BuyQ;  //<价格, 数目>,最大堆
        priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> SellQ;  //<价格, 数目>,最小堆
        for (auto& order : orders) {
            if (order[2] == 0) {   //当前订单是Buy,需要找最小的Sell
                while (order[1] > 0 && !SellQ.empty() && SellQ.top().first <= order[0]) {  //当前订单数目大于0,Sell最低价<= 当前订单价格
                    if (order[1] > SellQ.top().second) {  //当前订单数 > Sell最低价的订单数
                        order[1] -= SellQ.top().second;
                        SellQ.pop();  //Sell中最低价被删除
                    } else if (order[1] == SellQ.top().second) {  //当前订单数 == Sell最低价的订单数
                        SellQ.pop();  //Sell中最低价被删除
                        order[1] = 0;
                    } else if (order[1] < SellQ.top().second) { //当前订单数 < Sell最低价的订单数
                        auto minSell = SellQ.top(); SellQ.pop(); //更新Sell最低价的订单数量
                        SellQ.push(pair<int, int>(minSell.first, minSell.second - order[1]));
                        order[1] = 0;
                    }
                }
                if (order[1] > 0) BuyQ.push(pair<int, int>(order[0], order[1])); //当前订单数还有余留,则压入队列中
            } else {  //当前订单是Sell,需要找最大的Buy
                while (order[1] > 0 && !BuyQ.empty() && BuyQ.top().first >= order[0]) {  //当前订单数目大于0,Buy最高价>= 当前订单价格
                    if (order[1] > BuyQ.top().second) {  //当前订单数 > Buy最高价的订单数
                        order[1] -= BuyQ.top().second;
                        BuyQ.pop();  //Buy最高价被删除
                    } else if (order[1] == BuyQ.top().second) {  //当前订单数 == Buy最高价的订单数
                        BuyQ.pop();  //Buy最高价被删除
                        order[1] = 0;
                    } else if (order[1] < BuyQ.top().second) { //当前订单数 < Buy最高价的订单数
                        auto maxBuy = BuyQ.top(); BuyQ.pop(); //更新Buy最高价的订单数量
                        BuyQ.push(pair<int, int>(maxBuy.first, maxBuy.second - order[1]));
                        order[1] = 0;
                    }
                }
                if (order[1] > 0) SellQ.push(pair<int, int>(order[0], order[1])); //当前订单数还有余留,则压入队列中
            }
        }
        int res = 0;
        while (!BuyQ.empty()) {
            res = (res + BuyQ.top().second) % 1000000007;
            BuyQ.pop();
        }
        while (!SellQ.empty()) {
            res = (res + SellQ.top().second) % 1000000007;
            SellQ.pop();
        }
        return res;
    }
};

作者:victor3290
链接:https://leetcode-cn.com/problems/number-of-orders-in-the-backlog/solution/you-xian-dui-lie-da-xiao-ding-dui-by-vic-0b91/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

 

typedef pair<int, int> PII;

class Solution {
public:
    int getNumberOfBacklogOrders(vector<vector<int>>& orders) {
        priority_queue<PII> A, B;
        int MOD = 1e9 + 7;
        long long ans = 0;
        
        for(auto &i : orders){
            if(i[2] == 0)   //buy
                A.push({i[0], i[1]});
            else            //sell
                B.push({-i[0], i[1]});
            ans += i[1];
            //消除
            while(!A.empty()){
                if(B.empty() || -B.top().first > A.top().first)
                    break;
                PII x = A.top();
                A.pop();
                PII y = B.top();
                B.pop();
                
                int nu = min(x.second, y.second);
                x.second -= nu;
                y.second -= nu;
                ans -= 2 * nu;
                if(x.second)
                    A.push(x);
                else
                    B.push(y);
            }
            // cout << " " << ans << endl;
        }
        return ans % MOD;
    }
};

作者:Raymond_YP
链接:https://leetcode-cn.com/problems/number-of-orders-in-the-backlog/solution/c-you-xian-dui-lie-by-raymond_yp-7p1k/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值