线段树-详解

线段树

很多人都知道线段树的强大，也知道他是个什么样的结构，但一到用代码实现是不是就懵了！线段树模板其实就是个代码比较长，但代码重复很多的一个递归！

线段树用途：

数字之和——总数字之和 = 左区间数字之和 + 右区间数字之和

最大公因数(GCD)——总GCD = gcd( 左区间GCD , 右区间GCD );

最大值——总最大值=max(左区间最大值，右区间最大值)

学习自b站博主

下标从0开始的线段树

#include<bits/stdc++.h>//线段树 构建 
using namespace std;
const int MAX=1000;
int arr[MAX],tree[MAX]={0};

void pf(){
	for(int i=0;i<=16;i++){
		printf("tree[%d]=%d",i,tree[i]);
		cout << endl ;
	}
	cout << endl;
}
void build_tree(int arr[],int tree[],int node,int st,int ed){//构建完全二叉树 
	if(st>=ed){//递归出口 
		tree[node]=arr[st];//对递归到只有一个数的线段树赋值 
		return;
	}
	
	int mid=(st+ed)>>1; 
	int left_node=2*node+1;//计算此时线段树的坐标开始位置 
	int right_node=2*node+2;//下标从0开始和从1开始不一样，这里下标从推荐0开始 
	
	build_tree(arr,tree,left_node,st,mid);//左递归 
	build_tree(arr,tree,right_node,mid+1,ed);//右递归 
	tree[node]=tree[left_node]+tree[right_node];//将左右儿子相加 
}


void update_tree(int arr[],int tree[],int node,int st,int ed,int idx,int val){//更新数值O(logn) idx更改下标，val更改值 
	if(st==ed){
		arr[idx]=val;//修改原数组 
		tree[node]=val;//修改线段树中的值 
		return;
	}
	
	int mid=(st+ed)>>1;
	int left_node=node*2+1;
	int right_node=node*2+2;
	
	if(idx>=st&&idx<=mid) update_tree(arr,tree,left_node,st,mid,idx,val);//寻找修改的值在哪部分 
	else update_tree(arr,tree,right_node,mid+1,ed,idx,val);
	
	tree[node]=tree[left_node]+tree[right_node];//递归修改这条路上的值 
}

int query_tree(int arr[],int tree[],int node,int st,int ed,int L,int R){//区间最大值O(logn) 

	if(L>ed||R<st) return 0;//剪枝
	else if(st==ed) return tree[node];
	
	
	int mid=(st+ed)>>1;
	int left_node=node*2+1;
	int right_node=node*2+2;
	
	int sum_left=query_tree(arr,tree,left_node,st,mid,L,R);
	int sum_right=query_tree(arr,tree,right_node,mid+1,ed,L,R);
	int Max=max(sum_left,sum_right);
	Max=max(Max,sum_left+sum_right);
	return Max;
	
}


int main(){
	int n;
	cin >> n;
	for(int i=0;i<n;i++) cin >> arr[i];
	
	build_tree(arr,tree,0,0,n-1);//构建线段树 
	pf();
//	update_tree(arr,tree,0,0,n-1,4,6);//更新值 
//	pf();
//	cout << query_tree(arr,tree,0,0,n-1,0,n-1) << endl;
	return 0;
} 


/*
int query_tree(int arr[],int tree[],int node,int st,int ed,int L,int R){//区间最大值O(logn) 
	//用于检验查询的值 
//	printf("start=%d\n",st);
//	printf("end=  %d\n",ed);
//	cout << endl; 
	
	if(L>ed||R<st) return 0;//剪枝
	else if(st>=L&&ed<=R) return tree[node];//这步很重要，表示加上区间的值 
	else if(st==ed) return tree[node];//可有可无吧 
	
	
	int mid=(st+ed)>>1;
	int left_node=node*2+1;
	int right_node=node*2+2;
	
	int sum_left=query_tree(arr,tree,left_node,st,mid,L,R);
	int sum_right=query_tree(arr,tree,right_node,mid+1,ed,L,R);
	return sum_left+sum_right;//返回结果是相加的值 
	
}
*/