力扣 1801. 积压订单中的订单总数

news2024/12/27 16:27:39

题目

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

在这里插入图片描述
输入：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) 。

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/number-of-orders-in-the-backlog
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

方法1：优先队列

Java实现

class Solution {
    public int getNumberOfBacklogOrders(int[][] orders) {
        PriorityQueue<int[]> Buy_Backing = new PriorityQueue<>((a, b) -> {
            return b[0] - a[0];
        });
        PriorityQueue<int[]> Sell_Backing = new PriorityQueue<>((a, b) -> {
            return a[0] - b[0];
        });

        for (int[] order : orders) {
            if (order[2] == 0) { // buy
                while (order[1] > 0 && !Sell_Backing.isEmpty() && Sell_Backing.peek()[0] <= order[0]) {
                    int[] cur = Sell_Backing.poll();
                    if (cur[1] > order[1]) {
                        cur[1] -= order[1];
                        order[1] = 0;
                        Sell_Backing.offer(cur);
                    } else if (cur[1] < order[1]){
                        order[1] -= cur[1];
                    } else {
                        order[1] = 0;
                    }
                }
                if (order[1] > 0) Buy_Backing.offer(order);
            } else { // sell
                while (order[1] > 0 && !Buy_Backing.isEmpty() && Buy_Backing.peek()[0] >= order[0]) {
                    int[] cur = Buy_Backing.poll();
                    if (cur[1] > order[1]) {
                        cur[1] -= order[1];
                        order[1] = 0;
                        Buy_Backing.offer(cur);
                    } else if (cur[1] < order[1]) {
                        order[1] -= cur[1];
                    } else {
                        order[1] = 0;
                    }
                }
                if (order[1] > 0) Sell_Backing.offer(order);
            }
        }
        
        int res = 0;
        while (!Buy_Backing.isEmpty()) {
            res = (res + Buy_Backing.poll()[1]) % 1000000007;
        }
        while (!Sell_Backing.isEmpty()) {
            res = (res + Sell_Backing.poll()[1]) % 1000000007;
        }
        return res;
    }
}