线段树（累计和）

最新推荐文章于 2022-08-27 23:07:26 发布

OI之神

最新推荐文章于 2022-08-27 23:07:26 发布

阅读量59

点赞数 1

分类专栏：数据结构

本文链接：https://blog.csdn.net/qq_60821620/article/details/119706932

版权

数据结构专栏收录该内容

8 篇文章 0 订阅

订阅专栏

这篇博客展示了如何用C++编写一个区间查询和修改的数据结构，利用线段树实现。代码中详细定义了结构体T，并定义了build、pushdown、change和query等函数，用于维护区间加法并查集。博主分享了在处理query和change函数时遇到的问题，提醒读者注意避免类似错误。

摘要由CSDN通过智能技术生成

好气啊，因query和change的函数l，r返回值报错憋了5分钟，下次一定注意

long long暴力美学
这份代码写的还是很美观的

#include<bits/stdc++.h>
using namespace std;
const long long maxn=1e5+1;
long long n,m,a[maxn];
struct T
{
	long long data,add,l,r;
}node[4*maxn];
void build(long long pos,long long l,long long r)
{
	node[pos].l=l;
	node[pos].r=r;
	if(l==r)
	{
		node[pos].data=a[l];
		return;
	}
	long long mid=(l+r)>>1;
	build(pos<<1,l,mid);
	build(pos<<1|1,mid+1,r);
	node[pos].data=node[pos<<1].data+node[pos<<1|1].data;
}
void pushdown(long long pos)
{
	if(node[pos].add)
	{
		node[pos<<1].data+=node[pos].add*(node[pos<<1].r-node[pos<<1].l+1);
		node[pos<<1|1].data+=node[pos].add*(node[pos<<1|1].r-node[pos<<1|1].l+1);
		node[pos<<1].add+=node[pos].add;
		node[pos<<1|1].add+=node[pos].add;
		node[pos].add=0;
	}
}
void change(long long pos,long long l,long long r,long long v)
{
	if(l<=node[pos].l&&node[pos].r<=r)
	{
		node[pos].add+=v;
		node[pos].data+=v*(node[pos].r-node[pos].l+1);
		return;
	}
	pushdown(pos);
	long long mid=(node[pos].l+node[pos].r)>>1;
	if(l<=mid) change(pos<<1,l,r,v);
	if(r>mid) change(pos<<1|1,l,r,v);
	node[pos].data=node[pos<<1].data+node[pos<<1|1].data;
}
long long query(long long pos,long long l,long long r)
{
	if(l<=node[pos].l&&node[pos].r<=r)
	{
		return node[pos].data;
	}
	pushdown(pos);
	long long mid=(node[pos].l+node[pos].r)>>1,answer=0;
	if(l<=mid) answer+=query(pos<<1,l,r);
	if(r>mid) answer+=query(pos<<1|1,l,r);
	return answer;
}
int main()
{
	scanf("%lld%lld",&n,&m);
	memset(a,0,sizeof(a));
	for(long long i=1;i<=n;i++) scanf("%lld",&a[i]);
	build(1,1,n); 
	for(long long i=1;i<=m;i++)
	{
		long long a;
		scanf("%lld",&a);
		if(a==1)
		{
			long long b,c,d;
			scanf("%lld%lld%lld",&b,&c,&d);
			change(1,b,c,d);
		}
		else
		{
			long long b,c;
			scanf("%lld%lld",&b,&c);
			printf("%lld\n",query(1,b,c));
		}
	}
	return 0;
}