再识线段树之更新与查询

最新推荐文章于 2024-02-02 21:42:25 发布

忧郁三角龙ଲ

最新推荐文章于 2024-02-02 21:42:25 发布

阅读量112

点赞数

本文链接：https://blog.csdn.net/weixin_45894701/article/details/104439382

版权

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;//返回求和值
 }
}