线段树模板

题目链接
线段树模板

#include
#include
#include
#include
using namespace std;
const int maxn=100005;
int n,m,q;
long long a[maxn],lazy[4*maxn],ans[4*maxn];
inline long long ls(long long x) { return x<<1;   }
inline long long rs(long long x) { return x<<1|1; }
inline void push_up(long long x)
{
	ans[x]=ans[ls(x)]+ans[rs(x)];
}
inline void build(long long x,long long l,long long r)
{
	lazy[x]=0;
	if(l==r)
	{
		ans[x]=a[l];
		return;
	}
	long long mid=(l+r)>>1;
	build(ls(x),l,mid);
	build(rs(x),mid+1,r);
	push_up(x);
}
inline void push_down(long long x,long long l,long long r)
{
	long long mid=(l+r)>>1;
	lazy[ls(x)]+=lazy[x];
	lazy[rs(x)]+=lazy[x];
	ans[ls(x)]+=lazy[x]*(mid-l+1);
	ans[rs(x)]+=lazy[x]*(r-mid);
	lazy[x]=0;
}
inline void ins(long long x,long long al,long long ar,long long l,long long r,long long k)
{
	if(al<=l && r<=ar)
	{
		lazy[x]+=k;
		ans[x]+=k*(r-l+1);
		return;
	} 
	if(lazy[x]!=0) push_down(x,l,r);
	long long mid=(l+r)>>1;
	if(al<=mid) ins(ls(x),al,ar,l,mid,k);
	if(ar>mid)  ins(rs(x),al,ar,mid+1,r,k);
	push_up(x);
}
inline long long query(long long x,long long al,long long ar,long long l,long long r)
{
	long long sum=0;
	if(al<=l && r<=ar) return ans[x]; 
	if(lazy[x]!=0) push_down(x,l,r);
	long long mid=(l+r)>>1;
	if(al<=mid) sum+=query(ls(x),al,ar,l,mid);
	if(ar>mid)  sum+=query(rs(x),al,ar,mid+1,r);
	return sum;
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
	}
	build(1,1,n);
	long long l,r,k;
	while(m--)
	{
		scanf("%d",&q);
		if(q==1)
		{
			scanf("%lld%lld%lld",&l,&r,&k);
			ins(1,l,r,1,n,k);
		}
		if(q==2)
		{
			scanf("%lld%lld",&l,&r);
			printf("%lld\n",query(1,l,r,1,n));
		}
	}
}