线段树的概念及C语言实现.docxVIP

线段树的概念及C语言实现

一、线段树的概念

线段树是一种用于解决区间查询和区间更新问题的树形数据结构。它通过递归地将区间分割成更小的子区间，并构建一棵树来存储这些子区间的信息（如和、最大值、最小值等），从而实现对这些信息的快速查询和更新。

线段树的每个节点代表一个区间，并且每个节点的值都是其子节点所代表区间的某种特定值（如和、最大值等）的聚合。线段树的根节点代表整个区间，而叶子节点代表单个元素。

二、C语言实现

在C语言中实现线段树，首先需要定义节点结构体和相关函数。以下是一个简单的线段树实现，用于存储区间和并支持区间查询和单点更新。

1、定义节点结构体

#includestdio.h

#includestdlib.h

#defineMAXN100005//假设最大区间长度为100000

//线段树节点

typedefstructNode{

intstart,end;//当前节点代表的区间

intsum;//当前区间和

structNode*left,*right;//左右子节点

}Node;

//创建新节点

Node*newNode(intstart,intend){

Node*node=(Node*)malloc(sizeof(Node));

node-start=start;

node-end=end;

node-sum=0;

node-left=node-right=NULL;

returnnode;

}

2、构建线段树

//构建线段树

Node*buildTree(intarr[],intstart,intend){

if(startend)returnNULL;

Node*node=newNode(start,end);

if(start==end){//叶节点

node-sum=arr[start];

returnnode;

}

intmid=(start+end)/2;

node-left=buildTree(arr,start,mid);

node-right=buildTree(arr,mid+1,end);

//聚合子节点的信息

node-sum=node-left-sum+node-right-sum;

returnnode;

}

3、单点更新

//单点更新

voidupdate(Node*root,intindex,intvalue){

if(root-start==root-end){//叶节点

root-sum=value;

return;

}

intmid=(root-start+root-end)/2;

if(index=mid)

update(root-left,index,value);

else

update(root-right,index,value);

//更新当前节点的和

root-sum=root-left-sum+root-right-sum;

}

4、区间查询

//区间查询

intquery(Node*root,intstart,intend){

if(root-startend||root-endstart)

return0;

if(root-start=startroot-end=end)

returnroot-sum;

intmid=(root-start+root-end)/2;

intleftSum=query(root-left,start,end);

intrightSum=query(root-right,start,end);

returnleftSum+rightSum;

}

5、主函数示例

intmain(){

intarr[]={1,3,5,7,9,11};

intn=sizeof(arr)/sizeof(arr[0]);

Node*root=buildTree(arr,0,