线段树(模板)

本文介绍了树状数组(也称为线段树)的数据结构,用于高效地处理单点修改和区间查询操作。代码示例展示了三种不同场景的实现:单点修改+区间查询、区间修改+单点查询以及区间修改+区间查询。通过这些例子,读者可以理解树状数组如何通过分治策略快速更新和查询数组片段的信息。
摘要由CSDN通过智能技术生成

单点修改+区间查询

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

const int N = 1e5+10;
struct node{
	int l,r,sum;
}tree[N<<2];

void pushup(int k)
{
	tree[k].sum = tree[k<<1].sum + tree[k<<1|1].sum;
}

void build(int k, int l, int r)
{
	tree[k] = {l,r,0};
	if(l==r)
	{
		return ;
	}
	int mid = (l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	pushup(k);
}
void update(int k, int pos)
{
	if(tree[k].r==tree[k].l)
	{
		tree[k].sum = 1;
		return ;
	}
	int mid = (tree[k].l+tree[k].r)>>1;
	if(pos<=mid)
		update(k<<1,pos);
	else
		update(k<<1|1,pos);
	pushup(k);
}

int query(int k, int l, int r)
{
	if(tree[k].l>r||tree[k].r<l)
		return 0;
	if(tree[k].l>=l&&tree[k].r<=r)
		return tree[k].sum;
	return query(k<<1,l,r)+query(k<<1|1,l,r);
}

int main()
{
	
}

区间修改+单点查询

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

const int N = 1e5+10;
struct node{
	int l,r,sum;
}tree[N<<2];

void pushup(int k)
{
	tree[k].sum = tree[k<<1].sum + tree[k<<1|1].sum;
}

void build(int k, int l, int r)
{
	tree[k] = {l,r,0};
	if(l==r)
	{
		return ;
	}
	int mid = (l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	pushup(k);
}
void update(int k, int l, int r , int val)//区间修改 
{
	if(tree[k].l<r||tree[k].r>l)
		return;
	if(tree[k].l>=l&&tree[k].r<=r)
	{
		tree[k].sum +=val*(tree[k].r - tree[k].l+1);
		return ;
	}
	update(k<<1,l,r,val);
	update(k<<1|1,l,r,val);
	pushup(k);
}

int query(int k, int pos)//单点查询 
{
	if(tree[k].l==tree[k].r)
		return tree[k].sum;
	int mid = (tree[k].l+tree[k].r)>>1;
	if(pos<=mid)
		return query(k<<1,pos);
	else
		return query(k<<1|1,pos);
}
int main()
{
	
}

区间修改+区间查询

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

const int N = 1e5+10;
struct node{
	int l,r,sum,lazy;
}tree[N<<2];

void pushup(int k)
{
	tree[k].sum = tree[k<<1].sum + tree[k<<1|1].sum;
}

void pushdown(int k)
{
	if(tree[k].lazy)
	{
		int lz = tree[k].lazy;
		tree[k].lazy = 0;
		tree[k<<1].lazy+=lz;
		tree[k<<1|1].lazy+=lz;
		tree[k<<1].sum +=lz*(tree[k<<1].r-tree[k<<1].l+1);
		tree[k<<1|1].sum+=lz*(tree[k<<1].r-tree[k<<1|1]+1);
	}
}
void build(int k, int l, int r)
{
	tree[k] = {l,r,0};
	if(l==r)
	{
		return ;
	}
	int mid = (l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	pushup(k);
}
void update(int k, int l, int r , int val)//区间修改 
{
	if(tree[k].l<r||tree[k].r>l)
		return;
	if(tree[k].l>=l&&tree[k].r<=r)
	{
		tree[k].sum +=val*(tree[k].r - tree[k].l+1);
		return ;
	}
	pushdown(k);
	update(k<<1,l,r,val);
	update(k<<1|1,l,r,val);
	pushup(k);
}

int query(int k, int pos)//区间查询 
{
	if(tree[k].l >r || tree[k].r<l)
		return 0;
	if(tree[k].l>=l&&tree[k].r<=r)
		return tree[k].sum;
	pushdown(k);
	return query(k<<1,l,r)+query(k<<1|1,l,r);
}
int main()
{
	
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值