线段树详解（从大一视角来看）

0x01.线段树

For What?

具体问题是:求一个数组内任意L,R下标之间的所有数的和,而且要提供下标值的更新的功能,普通方法太慢了,所以我们引入tree[]数组来保存叶子节点和叶子节点和的数组,以减小复杂度为O(logn)

Function

1.build_tree Func

void build_tree(int arr[],int tree[],int node,int start,int end){
	if(start == end){//递归到叶节点
		tree[node] = arr[start];//叶节点赋值
	}
	else{//没到叶节点
		int mid = (start + end)/2;
		int left_node = 2 * node + 1;
		int right_node = 2 * node + 2;
		build_tree(arr,tree,left_node,start,mid);//递归创建左树
		build_tree(arr,tree,right_node,mid+1,end);//递归创建右树
		//左右递归完毕
		tree[node] = tree[left_node] + tree[right_node];
	}
	for(int i=0;i<15;i++)//查看树中的点 
	{
		printf("tree[%d]===%d\n",i,tree[i]);
	}
	printf("\n");
}

Function goal:将arr的数据放入各个叶节点和 节点和 的点

Core:递归调用,先向左分支,然后直到叶子节点,然后返回上层,然后叶子节点的上层递归进行下一条右分支的递归(在这个语境下也是叶子节点),然后上层递归进行下一条语句 求和,则叶节点的上层递归结束,返回上层递归的上层递归(对于上层递归的递归过程,叶节点的上层 叶子和 节点就可以看作是叶子节点上层的上层的"叶子节点")

Ps:

1.start和end本质上都是数组下标,则when start == end 则到叶子节点

2.left_node和right_node的计算方法,这是满二叉树的规律

2.update_tree Func

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];
	}
}

本质上与上一个函数没什么区别,传入参数多了两个 idx 目标下标 val 修改成val这个值

但是这里有个很重要的过程

注意:更新后,相关的tree_node也要更新,所以放在搜索之外(if else之外)

3.querry_tree Func

这个函数是最重要的,涉及到一个线段树的本质思想:剪枝降低复杂度,用空间换时间,必须要剪枝

//求arr[L]到arr[R]的和,重点是函数出口
int query_tree(int arr[],int tree[],int node,int start,int end,int L,int R){
	if(end<L||start>R){
		return 0;//进行一个剪枝
	}
	else if(L<=start&&R>=end){
		return tree[node];
	}
	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,mid+1,end,L,R);
		return sum_left+sum_right;
	}
}

剪枝

如代码所示,如果你任意递归求和过程中的任意子区间是你某次递归的(start和end的全集),直接返回当下递归调用的treenode叶子和就可以.

4.main Func

int main(){
	int arr[]={1,3,5,7,9,11}; 
	int size=6;
	int tree[MAX]={0};//存线段树 不要忘记清零,把虚点设为0补为完全二叉树 
	
	build_tree(arr,tree,0,0,size-1);
	for(int i=0;i<15;i++)//查看树中的点 
	{
		printf("tree[%d]===%d\n",i,tree[i]);
	}
	printf("\n");
 
	
	printf("***********************\n");
 
 
	update_tree(arr,tree,0,0,size-1,4,6);
	for(int i=0;i<15;i++)//查看树中的点 
	{
		printf("tree[%d]===%d\n",i,tree[i]);
	}
	printf("\n");
 
 
	printf("***********************\n");
	
 
	int s=query_tree(arr,tree,0,0,size-1,2,5);
	printf("s===%d\n",s);	//输出查询的区间和 
	
	
	printf("***********************\n");
	return 0;
}

没啥好说的,注意函数传值,仔细想想就会明白了