[每日一题]1801.积压订单中的订单总数
LeetCode.[1801. 积压订单中的订单总数] 给你一个二维整数数组 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 取余的结果。
Solution
class Solution {
public int getNumberOfBacklogOrders(int[][] orders) {
PriorityQueue<Integer> buyQue = new PriorityQueue<>((i1, i2)->orders[i2][0] - orders[i1][0]);
PriorityQueue<Integer> sellQue = new PriorityQueue<>((i1, i2)->orders[i1][0] - orders[i2][0]);
for(int i = 0; i < orders.length; i++){
int[] o = orders[i];
int type = o[2], price = o[0];
if(type == 0){ // o是采购订单buy
while(!sellQue.isEmpty() && o[1] > 0){
int[] q = orders[sellQue.peek()];
if(q[0] > price)break;
int min = Math.min(o[1], q[1]);
o[1] -= min;
q[1] -= min;
if(q[1] == 0)sellQue.poll();
}
if(o[1] > 0)buyQue.offer(i);
}else{ // o是销售订单 sell
while(!buyQue.isEmpty() && o[1] > 0){
int[] q = orders[buyQue.peek()];
if(q[0] < price)break;
int min = Math.min(o[1], q[1]);
o[1] -= min;
q[1] -= min;
if(q[1] == 0) buyQue.poll();;
}
if(o[1] > 0)sellQue.offer(i);
}
}
int ret = 0;
for(int[] o: orders)ret = (ret + o[1]) % (1000000007);
return ret;
}
}