【算法】【C++、珂朵莉树】珂朵莉树，以力扣715. Range 模块为例

原题链接

---

【算法】珂朵莉树，以力扣715. Range 模块为例

引言

在本文中，我们将探讨如何使用珂朵莉树（Chtholly Tree）来解决力扣715号问题——Range模块。这个问题要求我们设计一个数据结构，能够跟踪数字范围，并支持查询、添加和删除操作。

问题描述

Range模块需要实现以下三个主要方法：

1. void addRange(int left, int right)：添加半开区间 [left, right) 到跟踪范围，覆盖该区间内的所有数字。

2. boolean queryRange(int left, int right)：查询区间 [left, right) 内的每个数字是否都在跟踪中，若是则返回true，否则返回false。

3. void removeRange(int left, int right)：从跟踪范围中移除区间 [left, right) 内的所有数字。

解决思路

为了高效处理区间操作，我们选择使用珂朵莉树，这是一种基于有序集合，或者说红黑树的数据结构。每个节点表示一个区间，具有左边界（le）、右边界（ri）和值（val）。我们使用C++中的set来实现珂朵莉树。

珂朵莉树节点的结构体定义

对小于运算符的重载，保证了节点在set中的有序性
在声明变量val，也就是区间的值的时候，也要声明mutable，否则可能会CE，因为要进行区间赋值操作。

struct ChthollyNode {
    int le, ri;
    mutable int val;
    ChthollyNode(int l, int r, int v): le(l), ri(r), val(v) {}
    bool operator < (const ChthollyNode &other) const {return le < other.le;}
};

解决方案的详细步骤解析：

1. 初始化数据结构对象时，插入一个包含整个数据范围的节点，初始值为0。

tree.insert(ChthollyNode(1, 1e9, 0));

2. 实现split方法，用于在指定位置分割节点。寻找左边第一个le大于等于pos的节点，如果存在相等的节点，则直接返回该迭代器。否则，返回前一个le小于pos的节点，并在pos处分割。

set<ChthollyNode>::iterator split(int pos) {
    auto it = tree.lower_bound(ChthollyNode(pos, 0, 0));
    if (it != tree.end() && it->le == pos) return it;
    it--;
    int left = it->le, right = it->ri, value = it->val;
    tree.erase(it);
    tree.insert(ChthollyNode(left, pos - 1, value));
    return tree.insert(ChthollyNode(pos, right, value)).first;
}

3. 实现assign方法，用于将指定区间内的节点值全部设为给定值。首先，因为erase方法是左闭右开地进行区间删除，为了删除的时候能包含到right，因此在右侧位置（right+1）进行分割，然后在左侧位置（left）进行分割。接着，删除位于两个分割点之间的所有节点，最后插入新的节点。

void assign(int left, int right, int value) {
    auto end = split(right + 1);
    auto begin = split(left);
    tree.erase(begin, end);
    tree.insert(ChthollyNode(left, right, value));
}

4. 实现check方法，检查指定区间内的所有节点是否都具有给定值。通过遍历区间内的节点，如果有任何节点的值不符合要求，返回false，否则返回true。

bool check(int left, int right) {
    auto end = split(right + 1);
    auto begin = split(left);
    for (auto it = begin; it != end; it++) {
        if (it->val == 0) return false;
    }
    return true;
}

5. 实现构造函数，初始化数据结构对象时插入包含整个数据范围的节点，初始值为0。

RangeModule() {
    tree.insert(ChthollyNode(1, 1e9, 0));
}

6. 实现三个主要方法：addRange、queryRange和removeRange，分别调用assign和check方法。

void addRange(int left, int right) {
    assign(left, right - 1, 1);
}

bool queryRange(int left, int right) {
    return check(left, right - 1);
}

void removeRange(int left, int right) {
    assign(left, right - 1, 0);
}

完整代码

// 题目说白了就是：让你实现一个数据结构，queryRange用来检查区间[left, right)的数字是否被跟踪，addRange可以将区间[left, right)内的数字全部设置为“正在跟踪”，removeRange可以将区间[left, right)内的数字全部设置为“未跟踪”。

// 在这里使用“珂朵莉树”进行求解，设“1”为“正在跟踪”，设“0”为“未跟踪”


struct ChthollyNode {
    int le, ri;
    mutable int val;
    ChthollyNode(int l, int r, int v): le(l), ri(r), val(v) {}
    bool operator < (const ChthollyNode &other) const {return le < other.le;}
};

class RangeModule {
public:
    set<ChthollyNode> tree;
    set<ChthollyNode>::iterator split(int pos) {
        // 寻找左边第一个le >= pos 的节点
        auto it = tree.lower_bound(ChthollyNode(pos, 0, 0));
        // 已经有了这个le == pos的节点，直接返回！
        if (it != tree.end() && it -> le == pos) return it;
        it--; // 如果不是等于pos，那就一定是大于pos，取前一个 < pos的节点以便分割。
        int left = it -> le, right = it -> ri, value = it -> val;
        
        // 执行分割操作，简单直观。
               tree.erase(it);
               tree.insert(ChthollyNode(left, pos - 1, value));
        return tree.insert(ChthollyNode(pos, right, value)).first;
    }

    void assign(int left, int right, int value) {
        auto end = split(right + 1);
        auto begin = split(left);
        tree.erase(begin, end);
        tree.insert(ChthollyNode(left, right, value));
    }

    bool check(int left, int right) {
        auto end = split(right + 1);
        auto begin = split(left);
        for (auto it = begin; it != end; it++) {
            if (it -> val == 0) return false;
        }
        return true;
    }

    RangeModule() {
        // 数据范围是1 <= left < right <= 1e9
        tree.insert(ChthollyNode(1, 1e9, 0));
    }
    
    void addRange(int left, int right) {
        assign(left, right - 1, 1); // 因为是半开区间，也就是[left, right)，所以不包括right。
    }

    bool queryRange(int left, int right) {
        return check(left, right - 1);
    }
    
    void removeRange(int left, int right) {
        assign(left, right - 1, 0);
    }
};

/**
 * Your RangeModule object will be instantiated and called as such:
 * RangeModule* obj = new RangeModule();
 * obj->addRange(left,right);
 * bool param_2 = obj->queryRange(left,right);
 * obj->removeRange(left,right);
 */

总结

本文详细介绍了如何使用珂朵莉树解决力扣715号问题，包括数据结构的设计思路和每个方法的实现步骤。通过合理的数据结构选择和高效的区间操作，我们实现了一个能够有效跟踪数字范围的Range模块。