摘要：示例解題代碼注意，如果要在上提交解答，必須把接口和類的代碼一并提交，這里并沒有在寫類中

1.最經(jīng)典的線段樹問題：區(qū)間染色
有一面墻，長度為n，每次選擇一段墻進(jìn)行染色，m次操作后，我們可以看見多少種顏色？m次操作后，我們可以在[i, j]區(qū)間內(nèi)看見多少種顏色？

2.其他應(yīng)用場景
2017年注冊用戶中消費最高的用戶？消費最少的用戶？學(xué)習(xí)時間最長的用戶？

3.復(fù)雜度比較

4.線段樹原理圖
以求和為例：

1.基礎(chǔ)

線段樹不是完全二叉樹

線段樹是平衡二叉樹（整棵樹最大深度和最小深度最大差為1）

線段樹依然可以用數(shù)組表示

把不存在的節(jié)點看成null，線段樹即是完全二叉樹

2.數(shù)組存儲線段樹的空間需求

對滿二叉樹：

h層，一共有2^h-1個節(jié)點（大約2^h）

最后一層(h-1)層，有2^(h-1)個節(jié)點

最后一層的節(jié)點數(shù)大致等于前面所有層節(jié)點之和

如果需要存儲n個元素

n=2^k，只需要2n的空間

n=2^k+1，需要4n的空間

public class SegmentTree {

    private E[] data;
    private E[] tree;

    public SegmentTree(E[] arr) {
        data = (E[])new Object[arr.length];

        for (int i = 0; i < arr.length; i++) {
            data[i] = arr[i];
        }
        tree = (E[])new Object[4 * arr.length];
    }

    // 獲取元素個數(shù)
    public int getSize() {
        return data.length;
    }

    // 獲取某個索引上的值
    private E get(int index) {
        if (index < 0 || index >= data.length) {
            throw new IllegalArgumentException("index is illegal");
        }

        return data[index];
    }

    // 返回完全二叉樹的數(shù)組表示中，一個索引所表示的元素的左子樹的索引
    private int leftChild(int index) {
        return 2 * index + 1;
    }

    // 返回完全二叉樹的數(shù)組表示中，一個索引所表示的元素的右子樹的索引
    private int rightChild(int index) {
        return 2 * index + 2;
    }
}

1.定義融合器接口

public interface Merge {
    // 區(qū)間的元素如何定義由用戶決定
    E merge(E a, E b);
}

2.創(chuàng)建線段樹代碼

private Merge merge;

public SegmentTree(E[] arr, Merge merge) {

    // 線段樹的融合器，用于定義線段樹的區(qū)間元素到底如何存儲
    this.merge = merge;

    data = (E[])new Object[arr.length];

    for (int i = 0; i < arr.length; i++) {
        data[i] = arr[i];
    }
    tree = (E[])new Object[4 * arr.length];
    buildSegmentTree(0, 0, data.length - 1);
}

// 遞歸：在treeIndex的位置創(chuàng)建表示區(qū)間[l,r]的線段樹
private void buildSegmentTree(int treeIndex, int l, int r) {
    if (l == r) {
        tree[treeIndex] = data[l];
        return;
    }

    int leftTreeIndex = leftChild(treeIndex);
    int rightTreeIndex = rightChild(treeIndex);
    int min = (l + r) / 2;

    buildSegmentTree(leftTreeIndex, l, min);
    buildSegmentTree(rightTreeIndex, min + 1, r);
    tree[treeIndex] = merge.merge(tree[leftTreeIndex], tree[rightTreeIndex]);
}

@Override
public String toString() {

    StringBuilder res = new StringBuilder();

    res.append("[");
    for (int i = 0; i < tree.length; i++) {
        if (tree[i] == null) {
            res.append("null");
        } else {
            res.append(tree[i]);
        }

        if (i != tree.length - 1) {
            res.append(", ");
        }

    }
    res.append("]");

    return res.toString();
}

// 線段樹的查詢操作，區(qū)間[queryL, queryR]
public E query(int queryL, int queryR) {

   if (queryL < 0 || queryL >= data.length || queryR < 0 || queryR >= data.length) {
       throw new IllegalArgumentException("queryL or queryR is illegal");
   }

   return query(0, 0, data.length - 1, queryL, queryR);
}

// 遞歸，以treeIndex為根節(jié)點，區(qū)間為[l, r]，查詢區(qū)間為[queryL, queryR]
private E query(int treeIndex, int l, int r, int queryL, int queryR) {

   if (l == queryL && r == queryR) {
       return tree[treeIndex];
   }

   int leftChildIndex = leftChild(treeIndex);
   int rightChildIndex= rightChild(treeIndex);
   int mid = (l + r) / 2;

   if (queryL >= mid + 1) {
       return query(rightChildIndex, mid + 1, r, queryL, queryR);
   } else if (queryR <= mid) {
       return query(leftChildIndex, l, mid, queryL, queryR);
   }

   E left = query(leftChildIndex, l, mid, queryL, mid);
   E right = query(rightChildIndex, mid + 1, r, mid + 1, queryR);
   return merge.merge(left, right);
}

題目：303. 區(qū)域和檢索 - 數(shù)組不可變
描述：給定一個整數(shù)數(shù)組 nums，求出數(shù)組從索引 i 到 j (i ≤ j) 范圍內(nèi)元素的總和，包含 i, j 兩點。
示例：

給定 nums = [-2, 0, 3, -5, 2, -1]，求和函數(shù)為 sumRange()

sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3

解題代碼：

// 注意，如果要在leetcode上提交解答，必須把Merge接口和SegmentTree類的代碼一并提交，這里并沒有在寫NumArray類中
public class NumArray {

    private SegmentTree segTree;

    public NumArray(int[] nums) {

        if (nums.length > 0) {

            Integer[] data = new Integer[nums.length];
            for (int i = 0; i < nums.length; i++) {
                data[i] = nums[i];
            }

            segTree = new SegmentTree<>(data, (a, b) -> a + b);
        }
    }

    public int sumRange(int i, int j) {
        if (segTree == null) {
                throw new IllegalArgumentException("segment tree is null");
        }
        return segTree.query(i, j);
    }

}

public void set(int index, E e) {

    if (index < 0 || index >= data.length) {
        throw new IllegalArgumentException("index is illegal");
    }

    set(0, 0, data.length - 1, index, e);
}

private void set(int treeIndex, int l, int r, int index, E e) {
    if (l == r) {
        tree[treeIndex] = e;
        return;
    }

    int leftChildIndex = leftChild(treeIndex);
    int rightChildIndex= rightChild(treeIndex);
    int mid = (l + r) / 2;

    if (index >= mid + 1) {
        set(rightChildIndex, mid + 1, r, index, e);
    } else if (index <= mid) {
        set(leftChildIndex, l, mid, index, e);
    }

    tree[treeIndex] = merge.merge(tree[leftChildIndex], tree[rightChildIndex]);
}

題目：307. 區(qū)域和檢索 - 數(shù)組可修改
描述：給定一個整數(shù)數(shù)組 nums，求出數(shù)組從索引 i 到 j (i ≤ j) 范圍內(nèi)元素的總和，包含 i, j 兩點。
示例：

Given nums = [1, 3, 5]

sumRange(0, 2) -> 9
update(1, 2)
sumRange(0, 2) -> 8

解題代碼：

class NumArray {

    // 注意，如果要在leetcode上提交解答，必須把Merge接口和SegmentTree類的代碼一并提交，這里并沒有在寫NumArray類中    
    private SegmentTree segTree;
    
    public NumArray(int[] nums) {
        if (nums.length > 0) {

            Integer[] data = new Integer[nums.length];
            for (int i = 0; i < nums.length; i++) {
                data[i] = nums[i];
            }

            segTree = new SegmentTree<>(data, (a, b) -> a + b);
        }
    }

    public void update(int i, int val) {
        if (segTree == null) {
            throw new IllegalArgumentException("segment tree is null");
        }
        segTree.set(i, val);
    }

    public int sumRange(int i, int j) {
        if (segTree == null) {
            throw new IllegalArgumentException("segment tree is null");
        }
        return segTree.query(i, j);
    }
}