线段树(入门笔记)

版权声明:本文为博主原创文章,欢迎转载。如有问题,欢迎指正。 https://blog.csdn.net/weixin_42172261/article/details/88367758

什么是线段树
线段树是一种二叉搜索树,它将一个区间划分成一些单元区间,每个单元区间对应线段树中的一个叶结点。
对于线段树中的每一个非叶节点[a,b],它的左儿子表示的区间为[a,(a+b)/2],右儿子表示的区间为[(a+b)/2+1,b]。因此线段树是平衡二叉树,最后的子节点数目为N,即整个线段区间的长度。
在这里插入图片描述
定义结点
一般定义成结构体类型,其成员可以根据需要增加

struct node{
	int l;//左区间端点 
	int r;//右区间端点 
	int val;//此节点存值 
	int len;//子结点管理区间长度 
}tree[1005]; 

建树

void build(int cur, int l, int r)
{
	int mid=(l+r)/2;
	tree[cur].l=l, tree[cur].r=r;
	tree[cur].val=0, tree[cur].len=r-l+1;
	if (l==r)//只有到达叶结点时才把数组的值赋给它 
		tree[cur].val=arr[l];//arr为输入数据的数组
	else{
		build(cur*2, l, mid);
		build(cur*2+1, mid+1, r);
		tree[cur].val=tree[cur*2].val+tree[cur*2+1].val;//回溯过程中更新父结点 
	}
} 

单点更新

void updata(int cur, int id, int val)//cur为节点编号,id为修改数据的编号,val为增加的值 
{
	int l=tree[cur].l, r=tree[cur].r;
	int mid=(l+r)/2;
	if (l==r){
		tree[cur].val+=val;
		return;
	}
	if (id<=mid)
		updata(cur*2, id, val);
	else
		updata(cur*2+1, id, val);
	tree[cur].val=tree[cur*2].val+tree[cur*2+1].val;//回溯过程中更新父结点 
} 

区间查询

int query(int cur, int ql, int qr)
{
	int l=tree[cur].l, r=tree[cur].r;
	int mid=(l+r)/2, res=0;
	if (ql<=l && r<=qr)//如果此节点的区间在查询区间内,直接返回 
		return tree[cur].val;
	if (ql<=mid)
		res+=query(cur*2, ql, qr);
	if (qr>mid)//注意这里是两个if,不是分支关系 
		res+=query(cur*2+1, ql, qr);
	return res;
}

参考链接:
https://blog.csdn.net/zhhe0101/article/details/53871453
https://blog.csdn.net/weixin_40788897/article/details/81539106

猜你喜欢

转载自blog.csdn.net/weixin_42172261/article/details/88367758