线段树的应用

307、区域和检索-数组可修改

使用线段树解决区域和的问题

/**
 * 307、区域和检索-数组可修改
 * 线段树
 */
class NumArray {
    private int[] segmentTree;
    private int n;

    public NumArray(int[] nums) {
        n = nums.length;
        segmentTree = new int[nums.length * 4];
        build(0, 0, n - 1, nums);
    }

    public void update(int index, int val) {
        change(index, val, 0, 0, n - 1);
    }

    public int sumRange(int left, int right) {
        return range(left, right, 0, 0, n - 1);
    }
	
	/**
		构建线段树，nums里面的全部元素位于线段树的叶子结点，当s==e的时候说明当前范围只有一个节点，
		也就是找到了线段树的叶子结点，进行赋值。
		在构建的时候不需要对线段树的下标有过多的考虑，因为叶子结点会对应着当前nums的节点在线段树中的下标
		所以只看nums数组就可以。
	**/
    private void build(int node, int s, int e, int[] nums) {
        if (s == e) {
            segmentTree[node] = nums[s];
            return;
        }
        int m = s + (e - s) / 2;
        build(node * 2 + 1, s, m, nums);
        build(node * 2 + 2, m + 1, e, nums);
        segmentTree[node] = segmentTree[node * 2 + 1] + segmentTree[node * 2 + 2];
    }
	
	/**
     * 每次向下查找的时候，都会对node进行改变，nums和segmentTree叶子结点的元素是一一对应的，所以
     * 修改nums里面的元素，就是修改nums和segmentTree叶子节点的值，所以要满足start == end，
     * 然后再向上递归，修改父节点的值。
     * @param index
     * @param val
     * @param node
     * @param s
     * @param e
     */
    private void change(int index, int val, int node, int s, int e) {
        if (s == e) {
            segmentTree[node] = val;
            return;
        }
        int m = s + (e - s) / 2;
        // 比较要改变的节点和当前m的关系，如果在m左边，就去操作当前子树的左子树，否则操作当前子树的右子树。
        if (index <= m) {
            change(index, val, node * 2 + 1, s, m);
        } else {
            change(index, val, node * 2 + 2, m + 1, e);
        }
        segmentTree[node] = segmentTree[node * 2 + 1] + segmentTree[node * 2 + 2];
    }
	
	/**
     * 范围查询，如果当前的范围等于要查询的范围，直接返回线段树当前位置的值；
     * 如果right <= m，说明当前查询的范围在左子树，所以去左子树进行查询;
     * 如果left > m，说明查询范围在右子树，去右子树进行查询；
     * 否则当前查询的范围就是一部分在左子树，一部分在右子树，分开进行查询。
     * @param left
     * @param right
     * @param node
     * @param s
     * @param e
     */
    private int range(int left, int right, int node, int s, int e) {
        if (left == s && right == e) {
            return segmentTree[node];
        }
        int m = s + (e - s) / 2;
        // 说明当前要搜索的范围在当前子树的左子树，去左子树查询left到right范围的和。
        if (right <= m) {
            return range(left, right, node * 2 + 1, s, m);
        // 说明当前要搜索的范围在当前子树的右子树，去右子树查询left到right的区域和。
        } else if (left > m) {
            return range(left, right, node * 2 + 2, m + 1, e);
        } else {
        	// 说明当前要查询的范围左子树和右子树分别包含一部分，返回两部分之和。
            return range(left, m, node * 2 + 1, s, m) + range(m + 1, right, node * 2 + 2, m + 1, e);
        }
    }
}