【1801. 积压订单中的订单总数】
创始人
2024-05-04 09:19:09
0

来源:力扣(LeetCode)

描述:

给你一个二维整数数组 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:

1

输入:orders = [[10,5,0],[15,2,1],[25,1,1],[30,4,0]]
输出:6
解释:输入订单后会发生下述情况:
- 提交 5 笔采购订单,价格为 10 。没有销售订单,所以这 5 笔订单添加到积压订单中。
- 提交 2 笔销售订单,价格为 15 。没有采购订单的价格大于或等于 15 ,所以这 2 笔订单添加到积压订单中。
- 提交 1 笔销售订单,价格为 25 。没有采购订单的价格大于或等于 25 ,所以这 1 笔订单添加到积压订单中。
- 提交 4 笔采购订单,价格为 30 。前 2 笔采购订单与价格最低(价格为 15)的 2 笔销售订单匹配,从积压订单中删除这 2 笔销售订单。第 3 笔采购订单与价格最低的 1 笔销售订单匹配,销售订单价格为 25 ,从积压订单中删除这 1 笔销售订单。积压订单中不存在更多销售订单,所以第 4 笔采购订单需要添加到积压订单中。
最终,积压订单中有 5 笔价格为 10 的采购订单,和 1 笔价格为 30 的采购订单。所以积压订单中的订单总数为 6 。

示例 2:

2

输入:orders = [[7,1000000000,1],[15,3,0],[5,999999995,0],[5,1,1]]
输出:999999984
解释:输入订单后会发生下述情况:
- 提交 109 笔销售订单,价格为 7 。没有采购订单,所以这 109 笔订单添加到积压订单中。
- 提交 3 笔采购订单,价格为 15 。这些采购订单与价格最低(价格为 7 )的 3 笔销售订单匹配,从积压订单中删除这 3 笔销售订单。
- 提交 999999995 笔采购订单,价格为 5 。销售订单的最低价为 7 ,所以这 999999995 笔订单添加到积压订单中。
- 提交 1 笔销售订单,价格为 5 。这笔销售订单与价格最高(价格为 5 )的 1 笔采购订单匹配,从积压订单中删除这 1 笔采购订单。
最终,积压订单中有 (1000000000-3) 笔价格为 7 的销售订单,和 (999999995-1) 笔价格为 5 的采购订单。所以积压订单中的订单总数为 1999999991 ,等于 999999984 % (109 + 7) 。

提示:

  • 1 <= orders.length <= 105
  • orders[i].length == 3
  • 1 <= pricei, amounti <= 109
  • orderTypei 为 0 或 1

方法:优先队列模拟

根据题意,需要遍历数组 orders 中的订单并依次处理。对于遍历到的每个订单,需要找到类型相反的积压订单,如果可以匹配则执行这两笔订单并将积压订单删除,否则将当前订单添加到积压订单中。

由于寻找已有的积压订单时,需要寻找价格最高的采购订单或者价格最低的销售订单,因此可以使用两个优先队列分别存储积压的采购订单和积压的销售订单,两个优先队列称为采购订单优先队列和销售订单优先队列,分别满足队首元素是价格最高的采购订单和价格最低的销售订单。

遍历数组 orders,对于 order = [price, amount, orderType],执行如下操作。

  • 如果 orderType = 0,则表示 amount 个价格为 price 的采购订单,需要将这些采购订单和积压的销售订单匹配并执行。当销售订单优先队列中存在价格小于等于 price 的销售订单时,将当前采购订单和积压的销售订单匹配并执行,直到当前采购订单全部匹配执行、积压的销售订单全部匹配执行或者剩余积压的销售订单的价格都大于 price。如果还有剩余的当前采购订单尚未匹配执行,则将剩余的采购订单添加到采购订单优先队列中。
  • 如果 orderType = 1,则表示 amount 个价格为 price 的销售订单,需要将这些销售订单和积压的采购订单匹配并执行。当采购订单优先队列中存在价格大于等于 price 的采购订单时,将当前销售订单和积压的采购订单匹配并执行,直到当前销售订单全部匹配执行、积压的采购订单全部匹配执行或者剩余积压的采购订单的价格都小于 price。如果还有剩余的当前销售订单尚未匹配执行,则将剩余的销售订单添加到销售订单优先队列中。

遍历数组 orders 之后,两个优先队列中剩余的元素表示输入所有订单之后的积压订单,计算两个优先队列中剩余的订单总数并返回。

代码:

class Solution {
public:int getNumberOfBacklogOrders(vector>& orders) {const int MOD = 1000000007;priority_queue, vector>, less>> buyOrders;priority_queue, vector>, greater>> sellOrders;for (auto &&order : orders) {int price = order[0], amount = order[1], orderType = order[2];if (orderType == 0) {while (amount > 0 && !sellOrders.empty() && sellOrders.top().first <= price) {auto sellOrder = sellOrders.top();sellOrders.pop();int sellAmount = min(amount, sellOrder.second);amount -= sellAmount;sellOrder.second -= sellAmount;if (sellOrder.second > 0) {sellOrders.push(sellOrder);}}if (amount > 0) {buyOrders.emplace(price, amount);}} else {while (amount > 0 && !buyOrders.empty() && buyOrders.top().first >= price) {auto buyOrder = buyOrders.top();buyOrders.pop();int buyAmount = min(amount, buyOrder.second);amount -= buyAmount;buyOrder.second -= buyAmount;if (buyOrder.second > 0) {buyOrders.push(buyOrder);}}if (amount > 0) {sellOrders.emplace(price, amount);}}}int total = 0;while (!buyOrders.empty()) {total = (total + buyOrders.top().second) % MOD;buyOrders.pop();}while (!sellOrders.empty()) {total = (total + sellOrders.top().second) % MOD;sellOrders.pop();}return total;}
};

执行用时:176 ms, 在所有 C++ 提交中击败了97.94%的用户
内存消耗:57.1 MB, 在所有 C++ 提交中击败了82.47%的用户
复杂度分析
时间复杂度: O(nlogn),其中 n 是数组 orders 的长度。需要遍历数组 orders 一次,对于每个元素处理优先队列的时间是 O(logn),共需要 O(nlogn) 的时间,遍历结束之后计算剩余的积压订单总数需要 O(nlogn) 的时间。
空间复杂度:O(n),其中 n 是数组 orders 的长度。优先队列需要 O(n) 的空间。
author:LeetCode-Solution

相关内容

热门资讯

监控摄像头接入GB28181平... 流程简介将监控摄像头的视频在网站和APP中直播,要解决的几个问题是:1&...
Windows10添加群晖磁盘... 在使用群晖NAS时,我们需要通过本地映射的方式把NAS映射成本地的一块磁盘使用。 通过...
protocol buffer... 目录 目录 什么是protocol buffer 1.protobuf 1.1安装  1.2使用...
在Word、WPS中插入AxM... 引言 我最近需要写一些文章,在排版时发现AxMath插入的公式竟然会导致行间距异常&#...
【PdgCntEditor】解... 一、问题背景 大部分的图书对应的PDF,目录中的页码并非PDF中直接索引的页码...
Fluent中创建监测点 1 概述某些仿真问题,需要创建监测点,用于获取空间定点的数据࿰...
educoder数据结构与算法...                                                   ...
MySQL下载和安装(Wind... 前言:刚换了一台电脑,里面所有东西都需要重新配置,习惯了所...
修复 爱普生 EPSON L4... L4151 L4153 L4156 L4158 L4163 L4165 L4166 L4168 L4...
MFC文件操作  MFC提供了一个文件操作的基类CFile,这个类提供了一个没有缓存的二进制格式的磁盘...