(这个开始不会写,然后搜题解,看了一个SegementTree的方法,搜了搜B站【【数据结构】线段树(Segment Tree)】https://www.bilibili.com/video/BV1cb411t7AM?vd_source=7bb39de4d4811e58c4ce533810e765ef 学了学现在明白点了 写上了)
给你一个数组 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]
)- 方法优势:空间复杂度和时间复杂度都为logn级别,综合速度较快,不会增删等查询或者查询等增删.
- 思路:构建数组初始化数组用构造方法调用build方法,改数组,用updata方法调用change方法,用sumRange调用range方法
- 心理波动:选这个方法是因为之前刚学了个堆排序,练一练,那个tree的长度尽量越小越好,够用就行,我就从1测试到4,就4可以.前两天学了个左移右移听说比乘除法快 装个b用一下,算mid的时候用的
- class NumArray {
-
private int[] segmentTree;
private int n;
public NumArray(int[] nums) {
n = nums.length;
segmentTree = new int[nums.length*4];
build(0, 0, n - 1, nums);
}
public void update(int index, int val) {
change(index, val, 0, 0, n - 1);
}
public int sumRange(int left, int right) {
return range(left, right, 0, 0, n - 1);
}
private void build(int node, int start, int end, int[] nums) {//0,0,n-1,nums
if (start == end) {
segmentTree[node] = nums[start];
return;
}
else{
int mid=start + ((end - start) >>1);
int left_node=2*node+1;
int right_node=2*node+2;
build(left_node, start, mid, nums);
build(right_node, mid + 1, end, nums);
segmentTree[node] = segmentTree[left_node] + segmentTree[right_node];
}
}
private void change(int index, int val, int node, int start, int end) {
int mid=start + ((end - start) >>1);
int left_node=2*node+1;
int right_node=2*node+2;
if (start == end) {
segmentTree[node] = val;
return;
}
if (index <= mid&&index>=start) {
change(index, val, left_node, start, mid);
} else {
change(index, val, right_node, mid + 1, end);
}
segmentTree[node] = segmentTree[left_node] + segmentTree[right_node];
}
private int range(int left, int right, int node, int start, int end) {
if(left>end||right<start){
return 0;
}
else if(left<=start&&right>=end){
return segmentTree[node];
}
else if(start==end){
return segmentTree[node];
}
else{
int mid=start + ((end - start) >>1);
int left_node=2*node+1;
int right_node=2*node+2;
int sum_left=range(left,right,left_node,start,mid);
int sum_right=range(left,right,right_node,mid+1,end);
return sum_left+sum_right;
}
}
}
/**
* Your NumArray object will be instantiated and called as such:
* NumArray obj = new NumArray(nums);
* obj.update(index,val);
* int param_2 = obj.sumRange(left,right);
*/