#include<iostream>
using namespace std;
struct node{
int left,right;
int sum;
}tree[100];
int p,v;
void Build(int l,int r,int cur){
tree[cur].left=l;
tree[cur].right=r;
if(l==r){///到达单位区间
cin>>tree[cur].sum;///直接给sum赋值
return ;
}
int mid=(l+r)/2;
Build(l,mid,cur*2);///建立左儿子节点
Build(mid+1,r,cur*2+1);///建立右儿子节点
tree[cur].sum=tree[cur*2].sum+tree[cur*2+1].sum;///递归回去的时候计算各节点sum值
}
void Update(int cur){///原理和Build()一样,所以有的时候不写Build(),写Update就足够了
if(tree[cur].left==tree[cur].right&&tree[cur].left==p){///到达单位区间,并且是p所在的单位区间
tree[cur].sum=v;///直接更改sum值
return ;
}
int mid=(tree[cur].left+tree[cur].right)/2;
if(p<=mid)Update(cur*2);///p在左儿子节点
else Update(cur*2+1);///p在右儿子节点
tree[cur].sum=tree[cur*2].sum+tree[cur*2+1].sum;///递归回去的时候更新相关节点的sum值
}
int Query(int l,int r,int cur){
if(l==tree[cur].left&&r==tree[cur].right)///查询区间和节点的区间完全相同
return tree[cur].sum;///直接返回该节点的sum值
int mid=(tree[cur].left+tree[cur].right)/2;
if(r<=mid)return Query(l,r,cur*2);///查询区间完全在左儿子节点的区间,则查询左儿子节点
else if(l>mid)return Query(l,r,cur*2+1);///查询区间完全右儿子节点的区间,则查询右子节点
else return Query(l,mid,cur*2)+Query(mid+1,r,cur*2+1);///查询区间在左右儿子节点区间都有,则都查
}
int main(){
int n,l,r;
cin>>n;
Build(1,n,1);///建立区间[1,n],从根节点1开始
cin>>p>>v;///将p点的值改成v
Update(1);///从根节点1开始更新
cin>>l>>r;
cout<<Query(l,r,1);///查询区间[l,r],从根节点1开始
}
线段树的建立、更新、查找
最新推荐文章于 2022-12-12 19:32:55 发布