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

墨蓝 2023-10-16 15:52 46阅读 0赞

#### 目录 ####

*  1.题目
 *  2.思路
 *  3.代码实现（Java）

## 1.题目 ##

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

示例 1：

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

**示例 2：**

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

**提示：**  
1 <= flowers.length <= 5 \* 104  
flowers\[i\].length == 2  
1 <= starti <= endi <= 109  
1 <= people.length <= 5 \* 104  
1 <= people\[i\] <= 109

来源：力扣（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;
        }
    }

[9acc15070b284dd5990c70146dcbb0ad.png]: https://img-blog.csdnimg.cn/9acc15070b284dd5990c70146dcbb0ad.png
[ad7e523c8bf0497c9d90cf92561fb8a7.png]: https://img-blog.csdnimg.cn/ad7e523c8bf0497c9d90cf92561fb8a7.png