Java中的平衡樹
阿新 • • 發佈:2017-11-19
java 是否 起點 span int weekly 平衡樹 mar clas
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中的平衡樹