码迷,mamicode.com
首页 > 编程语言 > 详细

Java中的平衡树

时间:2017-11-19 14:17:52      阅读:196      评论:0      收藏:0      [点我收藏+]

标签:tco   int   lin   上界   amp   ceil   star   this   tps   

leetcode 729

给定一堆线段,每个线段都有一个起点、一个终点,用数组[(beg1,end1),(beg2,end2),(beg3,end3)......]来表示。可以提出以下问题:

  • 这些线段是否有重合部分
  • 这些线段的总长度是多少(多条线段重合部分只计算一次),这个问题相当于上面那个问题的加强版

如果一次性给定了这些线段,只需对这些线段进行排序即可O(nlog(n)),然后从左往右O(n)复杂度扫描一遍。

这道题要求在线算法。所以就需要对每条线段的start进行排序,当插入时,判断start的上界线段、下界线段是否可以容纳当前线段即可。如果用数组实现,每次插入需要O(n/2)的复杂度。如果用平衡树,每次插入需要O(log(n))的复杂度。

import java.util.TreeMap;

class MyCalendar {
class Node {
    int start;
    int end;

    Node(int start, int end) {
        this.start = start;
        this.end = end;
    }
}

TreeMap<Integer, Node> a = new TreeMap<>();

public MyCalendar() {
    a.put(-1, new Node(-1, 0));
    a.put((int) (1e9 + 7), new Node((int) (1e9 + 7), (int) (1e9 + 8)));
}

public boolean book(int start, int end) {
    Node floor = a.floorEntry(start).getValue();
    Node ceil = a.ceilingEntry(start).getValue();
    if (start >= floor.end && end <= ceil.start) {
        a.put(start, new Node(start, end));
        return true;
    } else {
        return false;
    }
}
}

Java中的平衡树

标签:tco   int   lin   上界   amp   ceil   star   this   tps   

原文地址:http://www.cnblogs.com/weidiao/p/7859566.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!