LeetCode:2276. 统计区间中的整数数目(TreeMap Java)
目录
2276. 统计区间中的整数数目
题目描述:
给你区间的 空 集,请你设计并实现满足要求的数据结构:
- 新增:添加一个区间到这个区间集合中。
- 统计:计算出现在 至少一个 区间中的整数个数。
实现 CountIntervals
类:
CountIntervals()
使用区间的空集初始化对象void add(int left, int right)
添加区间[left, right]
到区间集合之中。int count()
返回出现在 至少一个 区间中的整数个数。
注意:区间 [left, right]
表示满足 left <= x <= right
的所有整数 x
。
示例 1:
输入 ["CountIntervals", "add", "add", "count", "add", "count"] [[], [2, 3], [7, 10], [], [5, 8], []] 输出 [null, null, null, 6, null, 8] 解释 CountIntervals countIntervals = new CountIntervals(); // 用一个区间空集初始化对象 countIntervals.add(2, 3); // 将 [2, 3] 添加到区间集合中 countIntervals.add(7, 10); // 将 [7, 10] 添加到区间集合中 countIntervals.count(); // 返回 6 // 整数 2 和 3 出现在区间 [2, 3] 中 // 整数 7、8、9、10 出现在区间 [7, 10] 中 countIntervals.add(5, 8); // 将 [5, 8] 添加到区间集合中 countIntervals.count(); // 返回 8 // 整数 2 和 3 出现在区间 [2, 3] 中 // 整数 5 和 6 出现在区间 [5, 8] 中 // 整数 7 和 8 出现在区间 [5, 8] 和区间 [7, 10] 中 // 整数 9 和 10 出现在区间 [7, 10] 中
提示:
1 <= left <= right <= 109
- 最多调用
add
和count
方法 总计105
次 - 调用
count
方法至少一次
实现代码与解析:
TreeMap
class CountIntervals {
TreeMap<Integer, Integer> map = new TreeMap<>(); // 按左端点排序
int cnt = 0;
public CountIntervals() {
}
public void add(int left, int right) {
Map.Entry<Integer, Integer> it = map.floorEntry(right); // 小于等于right的最大键值对
while (it != null && it.getValue() >= left) { // 有重合,合并区间
int l = it.getKey(), r= it.getValue();
left = Math.min(left, l);
right = Math.max(right, r);
cnt -= r - l + 1; // 这区间的点暂时都去掉
map.remove(l);
it = map.floorEntry(right); // 再看是否重合,合并
}
// 跳出循环,说明没重合的了
cnt += right - left + 1; // 把合并好的区间的整数再加回来
map.put(left, right); // 把最终合并好的放入map中
}
public int count() {
return cnt;
}
}
原理思路:
map中存放区间,用左端点排序,没加入一个区间,就看是否有重复,循环合并区间,再计算区间中的数即可。
主要是学习TreeMap中的这些方法。
最近刚从C++改成用java刷题,还有点不习惯,好多方法也不太知道。