Range Sum Query - Mutable 区域和检索 - 数组可修改

给定一个整数数组  nums，求出数组从索引 i 到 j  (i ≤ j) 范围内元素的总和，包含 i,  j 两点。

update(i, val) 函数可以通过将下标为 i 的数值更新为 val，从而对数列进行修改。

示例:

Given nums = [1, 3, 5]

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

说明:

1. 数组仅可以在 update 函数下进行修改。
2. 你可以假设 update 函数与 sumRange 函数的调用次数是均匀分布的。

思路：

方法一：直接用brute的方法，从i加到j，具体代码我就不写了，总之如果面试遇到这样的问题这么回答肯定不行。时间复杂度：Time complexity : O(n)O(n) - range sum query, O(1)O(1) - update query

方法二：把原数组压缩成大小为sqrt(原数组大小)的新数组nums_this，在每个新数组中我们分三段来计算从i到j的值，分为startBlock，midBlock，endBlock，其中startBlock是i所在的新数组起始下标，endBlock是j所在的新数组的结束下标，剩下的跨过的区块就是midBlock的区块大小，那么这里单独计算就可以，除了startBlock和endBlock需要按照原数组一个一个值计算，midBlock可以压缩平方背计算。这里要注意的是下文代码中的len既指新压缩的数组长度，也代表新数组每个元素所映射的旧数组长度也是len。Time complexity : O(n)O(n) - preprocessing, O(\sqrt{n})O(n​) - range sum query, O(1)O(1) - update query

参考代码：

class NumArray {
public:
	vector<int> b;
	vector<int> nums_this;
	int len;
	NumArray(vector<int> nums) {
		nums_this = nums;
		len = ceil(sqrt(nums.size()));
		b.resize(len);
		for (int i = 0; i < nums.size(); i++) {
			b[i / len] += nums[i];
		}
	}

	void update(int i, int val) {
		b[i / b.size()] += (val - nums_this[i]);
		nums_this[i] = val;
	}

	int sumRange(int i, int j) {
		int res = 0;
		int startBlock = i / len;
		int endBlock = j / len;
		if (startBlock == endBlock) {
			for (int start = i; start <= j; start++) {
				res += nums_this[start];
			}
			return res;
		}
		for (int start = i; start < (startBlock+1)*len; start++) {
			res += nums_this[start];
		}
		for (int start = (startBlock + 1); start < endBlock; start++) {
			res += b[start];
		}
		for (int start = endBlock*len; start <= j; start++) {
			res += nums_this[start];
		}
		return res;
	}
};

/**
 * Your NumArray object will be instantiated and called as such:
 * NumArray obj = new NumArray(nums);
 * obj.update(i,val);
 * int param_2 = obj.sumRange(i,j);
 */