1.lc1801:积压订单中的订单总数(2023/1/2)
题目:
给你一个二维整数数组 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
取余的结果。
示例 :
输入: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 。
思路:【官方题解】
作者:力扣官方题解
链接:https://leetcode.cn/problems/number-of-orders-in-the-backlog/solutions/2039856/ji-ya-ding-dan-zhong-de-ding-dan-zong-sh-6g22/
根据题意,需要遍历数组 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<vector<int>>& orders) {
const int MOD = 1000000007;
priority_queue<pair<int, int>, vector<pair<int, int>>, less<pair<int, int>>> buyOrders;
priority_queue<pair<int, int>, vector<pair<int, int>>, greater<pair<int, int>>> 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;
}
};