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) -> 8Note:
- The array is only modifiable by the update function.
- You may assume the number of calls to update and sumRange function is distributed evenly.
Segment tree.
class SegTreeNode{
public:
SegTreeNode * left, * right;
int sum;
int leftidx, rightidx;
SegTreeNode():sum(0), leftidx(0), rightidx(0), left(nullptr), right(nullptr){}
};
class NumArray {
public:
SegTreeNode * root;
NumArray(vector<int> &nums) {
if(nums.size() == 0) return ;
root = buildSegmentTree(nums, 0, nums.size() - 1);
}
SegTreeNode* buildSegmentTree(vector<int>& nums, int start, int end){
SegTreeNode * node = new SegTreeNode();
node->leftidx = start;
node->rightidx = end;
int mid = (start + end) / 2;
if(start == end){
node->sum = nums[start];
}else{
node->left = buildSegmentTree(nums, start, mid);
node->right = buildSegmentTree(nums, mid + 1, end);
node->sum = node->left->sum + node->right->sum;
}
return node;
}
void update(int i, int val) {
_update(root, i, val);
}
void _update(SegTreeNode* node, int i, int val){
if(node->leftidx == node->rightidx && node->leftidx == i){
node->sum = val;
}else{
int mid = (node->leftidx + node->rightidx) / 2;
if(i <= mid){
_update(node->left, i, val);
}else{
_update(node->right, i, val);
}
node->sum = node->left->sum + node->right->sum;
}
}
int sumRange(int i, int j) {
return _sumRange(root, i, j);
}
int _sumRange(SegTreeNode* node, int i, int j){
if(node->leftidx > j || node->rightidx < i){
return 0;
}
else if(node->leftidx >= i && node->rightidx <= j){
return node->sum;
}else{
return _sumRange(node->left, i, j) + _sumRange(node->right, i, j);
}
}
};
// Your NumArray object will be instantiated and called as such:
// NumArray numArray(nums);
// numArray.sumRange(0, 1);
// numArray.update(1, 10);
// numArray.sumRange(1, 2);