leetcode 729. My Calendar I | 729. 我的日程安排表 I(线段树)

x33g5p2x  于2021-11-20 转载在 其他  
字(1.2k)|赞(0)|评价(0)|浏览(253)

题目

https://leetcode.com/problems/my-calendar-i/

题解

看了左神课之后,自己实现了下改造后的线段树(非常不优雅),因为数组放不下,所以自建 Node 结构。

class Node {
    int l;
    int r;
    boolean used; // 当前范围已全覆盖
    Node left;
    Node right;

    public Node(int l, int r) {
        this.l = l;
        this.r = r;
    }
}

class MyCalendar {
    Node info;

    public MyCalendar() {
        info = new Node(0, 1 << 30);
    }

    public boolean book(int start, int end) {
        if (canAdd(info, start, end - 1)) { // [左闭,右闭]区间
            add(info, start, end - 1);
            return true;
        } else {
            return false;
        }
    }

    public boolean canAdd(Node info, int L, int R) {
        if (info.used) return false;
        // 当前范围从未开垦且可以全覆盖,则不分发直接返回
        if (info.left == null && L <= info.l && info.right == null && R >= info.r - 1) return true;
        // 分发至左右子树
        int m = (info.l + info.r) / 2;
        if (L < m) {
            if (info.left != null && !canAdd(info.left, L, R)) return false;
        }
        if (R >= m) {
            if (info.right != null && !canAdd(info.right, L, R)) return false;
        }
        return true;
    }

    public void add(Node node, int L, int R) {
        if (L <= node.l && R >= node.r - 1) {
            node.used = true;
            return;
        }
        // 分发至左右子树
        int m = (node.l + node.r) / 2;
        if (L < m) {
            if (node.left == null) {
                node.left = new Node(node.l, m);
            }
            add(node.left, L, R);
        }
        if (R >= m) {
            if (node.right == null) {
                node.right = new Node(m, node.r);
            }
            add(node.right, L, R);
        }
    }
}

/** * Your MyCalendar object will be instantiated and called as such: * MyCalendar obj = new MyCalendar(); * boolean param_1 = obj.book(start,end); */

相关文章