线段树(线段树入门笔记)

什么是线段树
线段树是一种二叉搜索树,它将一个区间划分成一些单元区间,每个单元区间对应线段树中的一个叶结点。
对于线段树中的每一个非叶节点[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;
}

完整模板

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;

struct node{
	int l;
	int r;
	int val;
	int len;
	int lazy;
}tree [505];
int arr[505];

void pushup(int cur){
	tree[cur].val=tree[cur*2].val+tree[cur*2+1].val;
}

void pushdown(int cur){
	if (tree[cur].lazy!=0){
		tree[cur*2].lazy+=tree[cur].lazy;
		tree[cur*2+1].lazy+=tree[cur].lazy;
		tree[cur*2].val+=tree[cur*2].len*tree[cur].lazy;
		tree[cur*2+1].val+=tree[cur*2+1].len*tree[cur].lazy;
		tree[cur].lazy=0;
	}
}
//建树 
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;
	tree[cur].lazy=0;
	if (l==r)
		tree[cur].val=arr[l];
	else{
		build(cur*2, l, mid);
		build(cur*2+1, mid+1, r);
		pushup(cur);
	}
}

//单点更新  加值 
void update_point(int cur, int pos, 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;
		tree[cur].lazy+=val;
		return;
	}
	if (pos<=mid)
		update_point(cur*2, pos, val);
	else
		update_point(cur*2+1, pos, val);
	pushup(cur);
} 
//单点查询
int query_point(int cur, int pos){
	if (tree[cur].l==tree[cur].r)
		return tree[cur].val;
	pushdown(cur);
	
	int mid=(tree[cur].l+tree[cur].r)/2;
	if (pos<mid)
		query_point(cur*2, pos);
	else
		query_point(cur*2+1, pos);
} 
//区间查询
int query_interval(int cur, int ql, int qr){
	if (ql<=tree[cur].l && tree[cur].r<=qr)
		return tree[cur].val;
	pushdown(cur);
	
	int ans=0;
	int mid=(tree[cur].l+tree[cur].r)/2;
	if (ql<=mid)
		ans+=query_interval(cur*2, ql, qr);
	if (qr>mid)
		ans+=query_interval(cur*2+1, ql, qr);
	return ans; 
} 
//区间更新 加值 
void update_interval(int cur, int l, int r, int val){
	if (l<=tree[cur].l && tree[cur].r<=r){
		tree[cur].val+=(tree[cur].r-tree[cur].l+1)*val;
		tree[cur].lazy+=val;
		return;
	}
	pushdown(cur);
	
	int mid=(tree[cur].l+tree[cur].r)/2;
	if (l<=mid)
		update_interval(cur*2, l, r, val);
	if (r>mid)
		update_interval(cur*2+1, l, r, val);
	
	pushup(cur);
} 
int main(){
	
	return 0;
}

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

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

H4ppyD0g

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值