Leetcode 307. 区域和检索 - 数组可修改 线段树

地址 https://leetcode-cn.com/problems/range-sum-query-mutable/

给你一个数组 nums ，请你完成两类查询，其中一类查询要求更新数组下标对应的值，另一类查询要求返回数组中某个范围内元素的总和。
实现 NumArray 类：

NumArray(int[] nums) 用整数数组 nums 初始化对象
void update(int index, int val) 将 nums[index] 的值更新为 val
int sumRange(int left, int right) 返回子数组 nums[left, right] 的总和
（即，nums[left] + nums[left + 1], ..., nums[right]）

示例：
输入：
["NumArray", "sumRange", "update", "sumRange"]
[[[1, 3, 5]], [0, 2], [1, 2], [0, 2]]
输出：
[null, 9, null, 8]

解释：
NumArray numArray = new NumArray([1, 3, 5]);
numArray.sumRange(0, 2); // 返回 9 ，sum([1,3,5]) = 9
numArray.update(1, 2);   // nums = [1,2,5]
numArray.sumRange(0, 2); // 返回 8 ，sum([1,2,5]) = 8
 

提示：
1 <= nums.length <= 3 * 104
-100 <= nums[i] <= 100
0 <= index < nums.length
-100 <= val <= 100
0 <= left <= right < nums.length
最多调用 3 * 104 次 update 和 sumRange 方法

解法
线段树模板 单点修改区间和

class NumArray {
public:
	struct SegmentTree {
		int l, r;
		int sum;
	};
	vector<struct SegmentTree> t;

	void build(int p, int l, int r,const vector<int>& a) {
		t[p].l = l; t[p].r = r;
		if (l == r) { t[p].sum = a[l]; return; }
		int mid = l + r >> 1;
		build(p * 2, l, mid, a);
		build(p * 2 + 1, mid + 1, r, a);
		t[p].sum = t[p * 2].sum + t[p * 2 + 1].sum;
	}

	NumArray(vector<int>& nums) {
		t.resize(120010);
		nums.insert(nums.begin(),0);
		build(1, 1, nums.size()-1,nums);
	}

	void update(int p, int x, int v) {
		if (t[p].l == t[p].r) { t[p].sum = v; return; }
		int mid = t[p].l + t[p].r >> 1;
		if (x <= mid) { update(p * 2, x, v); }
		else { update(p * 2 + 1, x, v); }
		t[p].sum = t[p * 2].sum + t[p * 2 + 1].sum;
	}

	void update(int index, int val) {
		update(1,index+1,val);
	}

	int query(int p, int l, int r) {
		if (l <= t[p].l && r >= t[p].r) return t[p].sum;
		int mid = (t[p].l + t[p].r) / 2;
		int sum = 0;
		if (l <= mid) sum += query(p*2,l,r);
		if (r > mid) sum += query(p * 2 + 1, l, r);
		return sum;
	}

	int sumRange(int left, int right) {
		return query(1,left+1,right+1);
	}
};

我的视频题解空间