LeetCode_离散化差分_困难_2251.花期内花的数目

代码星辰

已于 2023-09-28 11:27:39 修改

阅读量263

点赞数

分类专栏： LeetCode 算法刷题文章标签： leetcode 离散化差分

于 2023-09-28 10:21:04 首次发布

本文链接：https://blog.csdn.net/weixin_43004044/article/details/133376960

版权

LeetCode 算法刷题专栏收录该内容

488 篇文章 32 订阅

订阅专栏

1.题目

给你一个下标从 0 开始的二维整数数组 flowers ，其中 flowers[i] = [start_i, end_i] 表示第 i 朵花的花期从 start_i 到 end_i （都包含）。同时给你一个下标从 0 开始大小为 n 的整数数组 people ，people[i] 是第 i 个人来看花的时间。请你返回一个大小为 n 的整数数组 answer，其中 answer[i] 是第 i 个人到达时在花期内花的数目。

示例 1：

在这里插入图片描述
输入：flowers = [[1,6],[3,7],[9,12],[4,13]], people = [2,3,7,11]
输出：[1,2,2,2]
解释：上图展示了每朵花的花期时间，和每个人的到达时间。
对每个人，我们返回他们到达时在花期内花的数目。

示例 2：

在这里插入图片描述
输入：flowers = [[1,10],[3,3]], people = [3,3,2]
输出：[2,2,1]
解释：上图展示了每朵花的花期时间，和每个人的到达时间。
对每个人，我们返回他们到达时在花期内花的数目。

提示：
1 <= flowers.length <= 5 * 10⁴
flowers[i].length == 2
1 <= starti <= endi <= 10⁹
1 <= people.length <= 5 * 10⁴
1 <= people[i] <= 10⁹

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/number-of-flowers-in-full-bloom

2.思路

（1）离散化差分

3.代码实现（Java）

//思路1————离散化差分
class Solution {
    public int[] fullBloomFlowers(int[][] flowers, int[] people) {
        TreeMap<Integer, Integer> treeMap = getDiffMap(flowers);
        int n = people.length;
        int[] res = new int[n];
        for (int i = 0; i < n; i++) {
            if (treeMap.containsKey(people[i])) {
                res[i] = treeMap.get(people[i]);
            } else {
                // lowerKey(key): 返回第一个小于 key 的键
                res[i] = treeMap.get(treeMap.lowerKey(people[i]));
            }
        }
        return res;
    }

    public TreeMap<Integer, Integer> getDiffMap(int[][] intervals) {
        TreeMap<Integer, Integer> map = new TreeMap<>();
        map.put(-1, 0);
        for (int[] interval : intervals) {
            map.put(interval[0], map.getOrDefault(interval[0], 0) + 1);
            map.put(interval[1] + 1, map.getOrDefault(interval[1] + 1, 0) - 1);
        }
        map.forEach((key, value) -> {
            if (key >= 0) {
                int preK = map.lowerKey(key);
                int preV = map.get(preK);
                value += preV;
                map.put(key, value);
            }
        });
        return map;
    }
}