LeetCode 307 Range Sum Query - Mutable（树状数组）

目标是求出   

Given an integer array nums, find the sum of the elements between indices i and j (i ≤ j), inclusive.

The update(i, val) function modifies nums by updating the element at index i to val.

Example:

Given nums = [1, 3, 5]

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

 本题利用树状数组，若用其他方式，则会发生超时错误，树状数组的时间复杂度为o(logn)

    树状数组的结构及实现见下网站

点击打开链接

   本代码中的init，初始化了一个c数组，即树状数组，其实现为每次都把元素写入到其应有的位置，相当于初始化的时候，每次都将当前nums的第i位元素更新至树状数组中的每个元素中。

  其初始化和更新的函数在原理上是一样的。

 i&-i 会返回树状数组应该叠加的元素的个数。

public class NumArray {

	public int[] nums;
	public int[] c;
	public int sum = 0;
	public int l ;
   public  NumArray(int[] nums) {
        this.nums = nums;
        l = nums.length;
        c = new int[l+1];
        for (int i = 0; i < l; i++) {
    		init(i,nums[i]);
    	}
    }
    
    public void init(int i, int val) {
	// TODO Auto-generated method stub
        i++;
		while (i <= l) {
			c[i] += val;
			i += lowb(i);//
		}
        
    }

	private int lowb(int i) {
		// TODO Auto-generated method stub
		return i&-i;
	}

	public void update(int i, int val) {
        
		int diff = val - nums[i];
		nums[i] = val;
		init(i, diff);
        
        
    }
    
    public int sumRange(int i, int j) {
    	return sum(j+1)-sum(i);
    }


	private int sum(int j) {
		int sum=0;
	    while(j>0)
	    {
	         sum+=c[j];
	         j=j-lowb(j);
	    }    
	    return sum;
	}

}