题目描述
给你一个二维整数数组 orders ,其中每个 orders[i] = [pricei, amounti, orderTypei] 表示有 amounti 笔类型为 orderTypei 、价格为 pricei 的订单。
订单类型 orderTypei 可以分为两种:
0表示这是一批采购订单buy1表示这是一批销售订单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 取余的结果。
示例 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:
输入: 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 <= 105orders[i].length == 31 <= pricei, amounti <= 109orderTypei为0或1
分析
因为要在处理订单的过程中频繁获取积压订单中价格最低的销售订单以及价格最高的采购订单, 涉及维护最值的问题,可以利用堆来实现。
因为要获取价格最低的销售订单,所以用小顶堆维护积压的销售订单,
因为要获取价格最高的采购订单,所以用大顶堆维护积压的采购订单。
有了以上两个堆之后,
小顶堆堆顶就维护了积压的价格最低的销售订单,
大顶堆堆顶就维护了积压的价格最高的采购订单。\
遍历输入订单数组,获取当前批订单的价格,数量以及类型。
如果当前是采购订单,当采购订单未被处理完成并且积压的价格最低的销售订单价格小于等于当前采购订单时,则匹配成功,处理两批订单,并维护积压订单数量。
如果当前是销售订单,当销售订单未被处理完成并且积压的价格最高的采购订单价格大于等于当前销售订单时,则匹配成功,处理两批订单,并维护积压订单数量。
最后处理完成所有订单,就获取到了积压订单总数,返回其对 10的9次方 + 7 取余的结果即可。
代码实现
// 封装通用堆类
class Heap {
// 传入compare函数区分大小顶堆
constructor(compare){
this.arr = [];
this.size = 0;
this.compare = compare;
}
// 插入操作
push(item){
this.arr.push(item);
this.size++;
// 插入后自下向上平衡调整
if(this.size>1){
let cur = this.size-1,
parent = (cur-1)>>1;
while(cur>0&&this.compare(this.arr[parent],this.arr[cur])){
[this.arr[parent],this.arr[cur]] =
[this.arr[cur],this.arr[parent]]
cur = parent,parent = (cur-1)>>1
}
}
}
// 弹出操作
pop(){
if(this.size === 1){
this.size = 0;
return this.arr.pop();
}
const res = this.arr[0];
this.arr[0] = this.arr.pop();
this.size--;
// 弹出后自上向下平衡调整
let cur = 0,
childl = 1,childr = 2;
while(
(childl<this.size&&this.compare(this.arr[cur],this.arr[childl])) ||
(childr<this.size&&this.compare(this.arr[cur],this.arr[childr]))
){
if(childr<this.size&&this.compare(this.arr[childl],this.arr[childr])){
[this.arr[cur],this.arr[childr]] =
[this.arr[childr],this.arr[cur]]
cur = childr
}else{
[this.arr[cur],this.arr[childl]] =
[this.arr[childl],this.arr[cur]]
cur = childl
}
childl = cur*2+1,childr = cur*2+2
}
// 返回弹出元素
return res;
}
// 获取堆顶元素
top(){
return this.arr[0]
}
}
var getNumberOfBacklogOrders = function(orders) {
// 创建大顶堆保存积压采购订单
const buyHeap = new Heap((a,b) => a[0]<b[0]),
// 创建小顶堆保存积压销售订单
sellHeap = new Heap((a,b) => a[0]>b[0]);
// 初始化积压订单总数为0
let res = 0;
// 遍历处理所有订单
for(let i = 0;i<orders.length;i++){
// 获取当前批订单价格,数量,类型
let [price,amount,type] = orders[i]
// 采购订单
if(type === 0){
// 当采购订单未被处理完成并且积压的价格最低的销售订单价格小于等于当前采购订单
// 则匹配成功,进行处理
while(amount&&sellHeap.size&&sellHeap.top()[0]<=price){
// 获取积压的价格最低的销售订单
const sellTop = sellHeap.pop();
// 更新积压订单总数
res -= sellTop[1]
// 如果销售订单数量大于采购订单数量
if(sellTop[1]>amount){
// 更新销售订单数量
sellTop[1] -= amount;
// 更新采购订单数量
amount = 0;
// 将剩余销售订单放入积压订单
sellHeap.push(sellTop)
// 更新积压订单总数
res += sellTop[1]
}else{
// 否则更新采购订单数量
amount -= sellTop[1]
}
}
// 如果当前批采购订单未被处理完成
if(amount){
// 将剩余采购订单放入积压订单
buyHeap.push([price,amount])
// 更新积压订单总数
res += amount
}
// 销售订单
}else{
// 当销售订单未被处理完成并且积压的价格最高的采购订单价格大于等于当前销售订单
// 则匹配成功,进行处理
while(amount&&buyHeap.size&&buyHeap.top()[0]>=price){
// 获取积压的价格最高的采购订单
const buyTop = buyHeap.pop();
// 更新积压订单总数
res -= buyTop[1]
// 如果采购订单数量大于销售订单数量
if(buyTop[1]>amount){
// 更新采购订单数量
buyTop[1] -= amount;
// 更新销售订单处理
amount = 0;
// 将剩余采购订单放入积压订单
buyHeap.push(buyTop)
// 更新积压订单总数
res += buyTop[1]
}else{
// 更新销售该订单数量
amount -= buyTop[1]
}
}
// 如果当前批销售订单未被处理完成
if(amount){
// 将剩余销售订单放入积压订单
sellHeap.push([price,amount])
// 更新积压订单总数
res += amount
}
}
}
// 返回积压订单总数对 10的9次方 + 7 取余的结果
return res%1000000007;
};
至此我们就完成了 leetcode-1801-积压订单中的订单总数