LeetCode:2276. 统计区间中的整数数目(TreeMap Java)

目录

2276. 统计区间中的整数数目

题目描述:

实现代码与解析:

TreeMap

原理思路:


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刷题,还有点不习惯,好多方法也不太知道。