731.我的日程安排表II

731. 我的日程安排表 II

题目：

实现一个 MyCalendar 类来存放你的日程安排。如果要添加的时间内不会导致三重预订时，则可以存储这个新的日程安排。

MyCalendar 有一个 book(int start, int end)方法。它意味着在 start 到 end 时间内增加一个日程安排，注意，这里的时间是半开区间，即 [start, end), 实数 x 的范围为，  start <= x < end。

当三个日程安排有一些时间上的交叉时（例如三个日程安排都在同一时间内），就会产生三重预订。

每次调用 MyCalendar.book方法时，如果可以将日程安排成功添加到日历中而不会导致三重预订，返回 true。否则，返回 false 并且不要将该日程安排添加到日历中。

请按照以下步骤调用MyCalendar 类: MyCalendar cal = new MyCalendar(); MyCalendar.book(start, end)

思路1：利用差分的思想

1、首先，map底层实现是红黑树，是一个平衡二叉树，所以是有序的结构，也就是遍历map得到的key值一定是递增（不降）的
2、如果start和end是10的9次方，数据量非常大，如果数据量比较小，比如说10的5次方范围内，则可以直接利用差分统计，映射到数组中，通过一次遍历得到的每个位置的预定重叠次数
3、数量大的话，原本映射到数组中的值，就映射到map中，由于map本身是有序的，同样可以采用差分+一次统计，统计过程中累加每个位置的值，再进行取最大值就是答案。
    4、利用差分数组的思想，每当我们预定一个新的日程安排[start,end) ，在start计数  cnt[start] +1，表示从start预定的数目加 1， 在end计数 cnt[end] -1，表示从 end 开始预定的数目减 1
    我们从起点的地方x，我们对计数进行累加依次求出预定数，如果预定数大于3的话就撤销此次的安排。

class MyCalendarTwo {
    map<int, int> cnt;
public:
    MyCalendarTwo() {
        cnt.clear();
    }
    
    bool book(int start, int end) {
        cnt[start]++;
        cnt[end]--;
        int sum = 0;
        for(auto iter = cnt.begin(); iter!=cnt.end(); iter++) {
            sum += iter->second;
            if (sum >= 3) {
                cnt[start]--;
                cnt[end]++;
                return false;
            }
        }
        return true;

    }
};

/**
 * Your MyCalendarTwo object will be instantiated and called as such:
 * MyCalendarTwo* obj = new MyCalendarTwo();
 * bool param_1 = obj->book(start,end);
 */

思路2:利用线段树

1、线段树的每个节点维护当前区间下的覆盖次数集合，用一个mask来表示，mask的第i为位1则表示这个区间下覆盖了i次的线段，再用lazy标记表示当前节点覆盖的区间被完全覆盖了lazy次；
2、然后提供两个借口，一个是查询接口（查询给定线段内的覆盖情况），一个是插入接口（对于给定线段执行一次插入
3、插入：
3.1 判断区间交集，如果没有交集直接返回
3.2假如当前节点不存在，则创建一个线段数节点
3.3将父节点的lazy更新到当前节点，并且根据lazy的值更新mask的值（执行左移lazy位的草嘴）；
3.4如果当前区间完全覆盖了线段数节点所表示的区间，则lazy字段自增，并且，mask左移一位，表示他所管辖的区间内的所有线段的覆盖次数都加了一次；
3.5递归计算左右子树，并且将当前接待您的lazy标记传递下去；
3.6遍历完毕以后，根据左右子树的mask值计算当前树的mask值，也就是将左右子树的mask值进行位或即可；
4、查询：
4.1如果当前节点为空，则判断lazy标记是否为0，为0则返回，否则创建一个新的节点
4.2把lazy标记应用到当前节点
4.3如果区间交集为空，则直接返回；
4.4如果区间完全覆盖树表示的区间，则将树节点的mask值位或到结果mask中；
4.5然后就是和插入的最后两步一样遍历左右子树，并且进行mask的合并

class MyCalendarTwo {
    //线段数每个节点维护当前区间下的覆盖次数的集合
    struct TreeNode {
        TreeNode *left, *right;
        int lazy; //表示当前节点覆盖的区间被完全覆盖了lazy次
        int mask; //1->覆盖0次 2->覆盖1次 4->覆盖2次 mask的第i位位1则表示这个区间下覆盖了i次的线段
        TreeNode() {
            lazy = 0;
            mask = 1;
            left = right = NULL;
        }
    };
    TreeNode *root;
public:
    MyCalendarTwo() {
        root = NULL;
    }
    //对于给定的线段执行一次插入 
    void insert(TreeNode*& tn, int l, int r, int start, int end, int lazy) {
        if (end < l || r < start) return; //判断区间交集，如果没有交集直接返回；
        if (tn == NULL) tn = new TreeNode(); //当前节点不存在，就创建一个线段树节点
        tn->lazy += lazy; //将父节点的lazy值更新到当前节点
        tn->mask <<= lazy; // 根据lazy的值更新mask的值（执行左移lzay位操作）

        if (start <= l && r <= end) { //当前区间完全覆盖了线段数节点所表示的区间
            tn->lazy ++; //lazy字段自增
            tn->mask <<= 1; //mask左移一位，表示他所管辖的区间内的所有线段的覆盖次数都增加了一次
            return ;
        }
        int nowlazy = tn->lazy;
        tn->lazy = 0;
        //递归遍历左右子树，将当前节点的lazy标记传递下去
        int mid = (l + r) >> 1;
        insert((tn->left), l, mid, start, end, nowlazy);
        insert((tn->right), mid+1, r, start, end, nowlazy);
        //遍历完毕以后，根据左右子树的mask值计算当前树的mask值，也就是将左右子树的mask值进行位或即可
        tn->mask = 0;
        if (tn->left) tn->mask|= tn->left->mask;
        if (tn->right) tn->mask |= tn->right->mask;

    }
    void query(TreeNode*& tn, int l, int r, int start, int end, int lazy, int& mask) {
        if (tn == NULL) {//如果当前节点为空
            if (lazy == 0) return; //判断lazy标记是否为0，为0返回
            tn = new TreeNode(); //否则创建一个新的节点
        }
        tn->lazy += lazy; //把lazy标记应用到当前节点
        tn->mask <<= lazy; //根据lazy的值更新mask的值（执行左移lzay位操作）

        if (end < l || r < start) return;//区间交集为空，直接但会
        if (start <= l && r <= end) { //区间完全覆盖树所表示的区间，将树节点的mask值位或到结果 mask中
            mask |= tn->mask;
            return;
        }

        int nowlazy = tn->lazy;
        tn->lazy = 0;
        //递归遍历左右子树，将当前节点的lazy值和mask值传递下去
        int mid = (r + l) >> 1;
        query((tn->left), l, mid, start, end, nowlazy, mask);
        query((tn->right), mid+1, r, start, end, nowlazy, mask);
        //遍历完毕以后，根据左右子树的mask值计算当前节点的mask值
        tn->mask = 0;
        if (tn->left) tn->mask |= tn->left->mask;
        if (tn->right) tn->mask |= tn->right->mask;

    }

    
    bool book(int start, int end) {
        int mask = 0;
        query(root, 0, 1000000000, start, end-1, 0, mask);
        if ((mask >> 2) >= 1) return false;
        insert(root, 0, 1000000000, start, end-1, 0);
        return true;

    }
};

/**
 * Your MyCalendarTwo object will be instantiated and called as such:
 * MyCalendarTwo* obj = new MyCalendarTwo();
 * bool param_1 = obj->book(start,end);
 */