Java中的平衡樹

標籤：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中的平衡樹