线段树模板题

线段树

题目链接洛谷

#include<iostream>
using namespace std;
const int maxn=4e5+10;
int n,m;
long long a[maxn]; 
struct SegmentTree
{
	long long sum;
	long long add;
	int l,r;
} s[maxn];
int build(int p,int l,int r)//建树
{
	s[p].add = 0;
	s[p].l=l,s[p].r=r;
	if(s[p].l==s[p].r)
	{
		s[p].sum=a[l];
		return 0;
	}
	int mid=(s[p].l+s[p].r)/2;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	s[p].sum=s[2*p].sum+s[2*p+1].sum;
};
void spread(int p)//push_down函数，将父节点的延迟标记赋给儿子
{
	if(s[p].add!=0)
	{
		s[p*2].sum+=s[p].add*(s[p*2].r-s[p*2].l+1);
		s[p*2+1].sum+=s[p].add*(s[p*2+1].r-s[p*2+1].l+1);
		s[p*2].add+=s[p].add;
		s[p*2+1].add+=s[p].add;
		s[p].add=0;
	}
	return;
}
void change(int p,int tag,int l,int r,int v)
{
	if(l<=s[p].l&&r>=s[p].r)//如果这段线段被完全覆盖。
	{
		s[p].add+=v;
		s[p].sum+=v*(s[p].r-s[p].l+1);	
		return;
	}
	spread(p);
	int mid=(s[p].l+s[p].r)/2;
	if(l<=mid)change(p*2,tag,l,r,v);
	if(r>mid)change(p*2+1,tag,l,r,v);
	s[p].sum=s[p*2].sum+s[p*2+1].sum;
}
long long query(int p,int l,int r)
{
	if(l<=s[p].l&&r>=s[p].r)
	{
		return s[p].sum;
	}
	spread(p);
	int mid=(s[p].l+s[p].r)/2;
	long long ans=0;
	if(l<=mid)ans+=query(p*2,l,r);
	if(r>mid)ans+=query(p*2+1,l,r);
	return ans;
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	build(1,1,n);
	int tag,l,r,v;
	for(int i=0;i<m;i++)
	{
		cin>>tag>>l>>r;
		if(tag==2)cout<<query(1,l,r)<<endl;
		else 
		{
			cin>>v;
			change(1,tag,l,r,v);
		}
	}
	return 0;
}