线段树

来源：https://www.luogu.org/problemnew/show/P3372

 

线段树详解：https://user.qzone.qq.com/50222268/blog/1500000480

 

代码啦啦啦~

 

注意add向下传递后要清零哟

 

#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

struct Node
{
	int left,right;
	long long add,sum;
};

const int maxSize=100000;
Node node[maxSize*4+5];
long long a[maxSize+5];
long long sum,n,m;

void pushup(int u)
{
	node[u].sum=node[2*u].sum+node[2*u+1].sum;	//左右子树和相加 
	return ;
}

void build(int u,int begin,int end)
{
	int mid;
	
	node[u].left=begin;		node[u].right=end;
	if (begin==end)
	{
		node[u].sum=a[begin];
		return ;
	}
	
	mid=(begin+end)/2;
	build(2*u,begin,mid);
	build(2*u+1,mid+1,end);
	pushup(u);
}

void update(int u,int begin,int end,int k)
{
	int mid;
	
	if (node[u].left==begin && node[u].right==end)
	{
		node[u].sum+=k*(end-begin+1);
		node[u].add+=k;			//更新数据 
		return ;
	}
	node[u].sum+=k*(end-begin+1);	//更新当时的点 
	
	if (node[u].add!=0)		//后面的节点还没更新
	{
		node[2*u].add+=node[u].add;
		node[2*u+1].add+=node[u].add;
		node[2*u].sum+=node[u].add*(node[2*u].right-node[2*u].left+1);
		node[2*u+1].sum+=node[u].add*(node[2*u+1].right-node[2*u+1].left+1);		
		node[u].add=0;		//注意清零 
	}
	
	mid=(node[u].left+node[u].right)/2;
	if (end<=mid)
		update(2*u,begin,end,k);
	else if (begin>mid)
		update(2*u+1,begin,end,k);
	else
	{
		update(2*u,begin,mid,k);
		update(2*u+1,mid+1,end,k);
	}
}

void query(int u,int begin,int end)
{
	int mid;
	
	if (node[u].left==begin && node[u].right==end)
	{
		sum+=node[u].sum;
		return ;
	}
	
	if (node[u].add!=0)		//后面的节点还没更新
	{
		node[2*u].add+=node[u].add;
		node[2*u+1].add+=node[u].add;
		node[2*u].sum+=node[u].add*(node[2*u].right-node[2*u].left+1);
		node[2*u+1].sum+=node[u].add*(node[2*u+1].right-node[2*u+1].left+1);	
		node[u].add=0;	
	}
	
	mid=(node[u].left+node[u].right)/2;
	if (mid>=end)
		query(2*u,begin,end);
	else if (mid<begin)
		query(2*u+1,begin,end);
	else
	{
		query(2*u,begin,mid);
		query(2*u+1,mid+1,end);
	}
	
}

int main()
{
	long long i,k,x,y,z;
	
	freopen("a.txt","r",stdin);
	scanf("%lld%lld",&n,&m);
	
	for (i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	
	build(1,1,n);
	
	for (i=0;i<m;i++)
	{
		scanf("%lld",&k);
		if (k==1)
		{
			scanf("%lld%lld%lld",&x,&y,&z);
			update(1,x,y,z);
		}
		else
		{
			scanf("%lld%lld",&x,&y);
			sum=0;
			query(1,x,y);
			printf("%lld\n",sum);
		}
	}
	
	return 0;
}