1. 程式人生 > >Java中的平衡樹

Java中的平衡樹

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