标签:检索 node end nums int 307 start 数组 left
https://leetcode-cn.com/problems/range-sum-query-mutable/
给你一个数组 nums ,请你完成两类查询。
其中一类查询要求 更新 数组nums下标对应的值
另一类查询要求返回数组nums中索引left和索引right之间(包含)的nums元素的 和,其中left <= right
实现 NumArray 类:
NumArray(int[] nums) 用整数数组 nums 初始化对象
void update(int index, int val) 将 nums[index] 的值 更新 为 val
int sumRange(int left, int right) 返回数组nums中索引left和索引right之间(包含)的nums元素的 和(即,nums[left] + nums[left + 1], ..., nums[right])
示例 1:
输入:
["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); // 返回 1 + 3 + 5 = 9
numArray.update(1, 2); // nums = [1,2,5]
numArray.sumRange(0, 2); // 返回 1 + 2 + 5 = 8
方法一:线性树
class NumArray {
int[] nums;
int[] tree;
int n;
public NumArray(int[] nums) {
this.nums = nums;
n = nums.length;
if (n == 0) {
return;
}
tree = new int[n * 4];
buildTree(0, 0, n - 1);
}
void buildTree(int node, int start, int end) {
if (start == end) {
tree[node] = nums[start];
return;
}
int mid = (end + start) / 2 + start;
int left = node * 2 + 1;
int right = node * 2 + 2;
buildTree(left, start, mid);
buildTree(right, mid + 1, end);
tree[node] = tree[left] + tree[right];
}
public void update(int index, int val) {
updateTree(index, 0, val, 0, n - 1);
}
public void updateTree(int idx, int node, int val, int start, int end) {
if (start > end) {
return;
}
if (start == end) {
nums[idx] = val;
tree[node] = val;
} else {
int mid = (start + end) >> 1;
int left = node * 2 + 1;
int right = node * 2 + 2;
if (idx >= start && idx <= mid) {
updateTree(idx, left, val, start, mid);
} else {
updateTree(idx, right, val, mid + 1, end);
}
tree[node] = tree[left] + tree[right];
}
}
public int sumRange(int left, int right) {
return query(left, right, 0, 0, n - 1);
}
public int query(int l, int r, int node, int start, int end) {
if (l > end || r < start) {
return 0;
}
if (start == end) {
return tree[node];
}
if (l <= start && r <= end) {
return tree[node];
} else {
int mid = (start + end) >> 1;
int left = node * 2 + 1;
int right = node * 2 + 2;
int ls = query(l, r, left, start, mid);
int rs = query(l, r, right, end, mid + 1);
return ls + rs;
}
}
}
标签:检索,node,end,nums,int,307,start,数组,left 来源: https://www.cnblogs.com/liufei2/p/16100904.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。