6066. 统计区间中的整数数目（区间并集模板、动态开点）

ღCauchyོꦿ࿐

已于 2022-05-15 22:09:22 修改

阅读量156

点赞数 3

分类专栏：力扣周赛文章标签： leetcode 动态规划算法

于 2022-05-15 21:43:15 首次发布

本文链接：https://blog.csdn.net/qq_52678569/article/details/124788820

版权

力扣周赛专栏收录该内容

20 篇文章 2 订阅

订阅专栏

本文介绍了两种不同的数据结构实现，动态开点和使用set集合结合二分查找，来高效地处理区间计数的问题。动态开点通过递归划分区间并更新节点信息，而set集合方案则利用二分查找快速定位相交区间并进行合并。这两种方法都是为了解决在大量区间中统计被覆盖整数个数的高效算法。

摘要由CSDN通过智能技术生成

文章目录

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

题意

在这里插入图片描述

样例

在这里插入图片描述

数据范围

在这里插入图片描述

代码

动态开点

class CountIntervals {
    CountIntervals *left = nullptr, *right = nullptr;
    int l, r, sum = 0;
public:
    CountIntervals() :l(1), r(1e9) {}
    CountIntervals(int l, int r) :l(l), r(r) {}
    
    void add(int L, int R) {
        if(sum >= r - l + 1) return ; // 到该已经覆盖完的节点，不用向下走了
        if(l >= L && r <= R) { sum = r - l + 1; return ; } // 该节点能被完全覆盖
        int mid = (l + r) >> 1;
        if(left == nullptr) left = new CountIntervals(l, mid);
        if(right == nullptr) right = new CountIntervals(mid + 1, r);
        if(L <= mid) left->add(L, R);
        if(R > mid) right->add(L, R);
        sum = left->sum + right->sum;
    }
    
    int count() { return sum; }
};

/**
 * Your CountIntervals object will be instantiated and called as such:
 * CountIntervals* obj = new CountIntervals();
 * obj->add(left,right);
 * int param_2 = obj->count();
 */

set集合，二分

class CountIntervals {
public:
    #define PII pair<int, int>
    set<PII> s; // <R, L>，存的是R，L
    int res = 0;

    CountIntervals() {}
    
    // 以右端点排
    void add(int left, int right) {
        int l = left, r = right;
        // 找到第一个右端点>l， 即从相交的开始。
        auto it = s.lower_bound(PII{l, -1e9});
        // 向后找到第一个左端点>r，即不在相交了
        while(it != s.end()) {
            if(it->second > r) break;
            // 维护最终合并完的最小l，最大r
            l = min(it->second, l);
            r = max(it->first, r);
            // 减去所有找到的区间值
            res -= it->first - it->second + 1;
            s.erase(it++);
        }
        // 加上最终合并的区间值
        res += r - l + 1;
        // 在插入当前合并的区间
        s.insert(PII{r, l});
    }
    
    int count() { return res; }
};

/**
 * Your CountIntervals object will be instantiated and called as such:
 * CountIntervals* obj = new CountIntervals();
 * obj->add(left,right);
 * int param_2 = obj->count();
 */