1.更新操作:
原题目如图所示:给出一组数1,3,5,7,9,11,输出下标对应数字。
若将原数组中第五格的数字9更新替换为数字6,则需进行线段树之更新操作,更新后应如下图所示:
代码如下:
void update_tree(int arr[],int tree[],int node,int start,int end,int idx,int val){
if(start==end){//设置出口
arr[idx]=val;
tree[node]=val;
}
else{
int mid = (start+end)/2;
int left_node=2*node+1;
int right_node=2*node+2;
if(idx>=start&&idx<=mid){//在左边范围
update_tree(arr,tree,left_node,start,mid,idx,val);
}//左递归
else{//在右边范围
update_tree(arr,tree,right_node,mid+1,end,idx,val);
}//右递归
tree[node]=tree[left_node]+tree[right_node];//递归求和
}
}
2.查询操作:
需要用到query函数,query(L,R)即查询到区间(L,R)并求和。
代码如下:
int query_tree(int arr[],int tree[],int node,int start,int end,int L,int R){
if(R<start||L>end){
return 0;//所求和的区间不在范围内
}else if
(start==end){
return tree[node];
}else{
int mid = (start+end)/2;
int left_node=2*node+1;
int right_node=2*node+2;
int sum_left=query_tree(arr,tree,left_node,start,mid,L,R);//左边递归求和
int sum_right=query_tree(arr,tree,right_node,start,mid,L,R);//右边递归求和
return sum_left+sum_right;//返回求和值
}
}