目录
- 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:
输入: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 * 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;}
}