1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
| class NumArray { int n; int[] tree, nums;
int lowbit(int x) { return x & -x; } int query(int x) { int ans = 0; for (int i = x; i > 0; i -= lowbit(i)) ans += tree[i]; return ans; } void add(int x, int u) { for (int i = x; i <= n; i += lowbit(i)) tree[i] += u; } void init(int[] nums) { tree = new int[nums.length + 1]; for (int i = 0; i < nums.length; ++i) add(i + 1, nums[i]); }
public NumArray(int[] nums) { this.nums = nums; n = nums.length; init(nums); } public void update(int index, int val) { add(index + 1, val - nums[index]); nums[index] = val; } public int sumRange(int left, int right) { return query(right + 1) - query(left); } }
|