Segment Tree Template

最新推荐文章于 2020-07-24 10:29:38 发布

Unisolate

最新推荐文章于 2020-07-24 10:29:38 发布

阅读量883

点赞数 1

CC 4.0 BY-SA版权

分类专栏： acm

本文链接：https://blog.youkuaiyun.com/unisolate/article/details/19070621

acm 专栏收录该内容

18 篇文章

订阅专栏

本文详细介绍了区间树状数组的数据结构实现，包括构建、更新单点及区间、查询区间求和或最大值等核心操作，并提供了完整的代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#define lson l,m,n<<1
#define rson m+1,r,n<<1|1

int num[MAXN],sum[MAXN<<2],max[MAXN<<2],add[MAXN<<2];

void pushUp(int n)
{
	sum[n]=sum[n<<1]+sum[n<<1|1];
	max[n]=max(max[n<<1],max[n<<1|1]);
}

void build(int l,int r,int n)
{
	add[n]=0;
	if(l==r)
	{
		scanf("%d",&num[n]);
		sum[n]=max[n]=num[n];
		return;
	}
	int m=(l+r)>>1;
	build(lson);
	build(rson);
	pushUp(n);
}

//update only one point
void update(int x,int v,int l,int r,int n)
{
	if(l==r)
	{
		//oprate here
		return;
	}
	int m=(l+r)>>1;
	if(x<=m) //Very Important
		update(x,v,lson);
	else update(x,v,rson);
	pushUp(n);
}

//get sum or max val of [L,R]
int query(int L,int R,int l,int r,int n)
{
	if(L<=l&&r<=R)
	{
		return sum[n];
		// return max[n];
	}
	int ans=0,m=(l+r)>>1;
	if(L<=m)
		ans+=query(L,R,lson);
		// ans=max(ans,query(L,R,lson));
	if(m<R)
		ans+=query(L,R,rson);
		// ans=max(ans,query(L,R,rson));
	return ans;
}

void pushDown(int n,int m)
{
	if(add[n])
	{
		add[n<<1]+=add[n];
		add[n<<1|1]+=add[n];
		sum[n<<1]+=add[n]*(m-(m>>1));
		sum[n<<1|1]+=add[n]*(m>>1);
		add[n]=0;
	}
}

//update [L,R]
void update(int L,int R,int v,int l,int r,int n)
{
	if(L<=l&&r<=R)
	{
		add[n]+=v;
		sum[n]+=v*(r-l+1);
		return;
	}
	pushDown(n,r-l+1);
	int m=(l+r)>>1;
	if(L<=m)
		update(L,R,v,lson);
	if(m<R)
		update(L,R,v,rson);
	pushUp(n);
}

int query(int L,int R,int l,int r,int n)
{
	if(L<=l&&r<=R)
	{
		return sum[n];
	}
	pushDown(n,r-l+1);
	int m=(l+r)>>1;
	int ans=0;
	if(L<=m)
		ans+=query(L,R,lson);
	if(m<R)
		ans+=query(L,R,rson);
	return ans;
}