A Simple Problem with Integer（线段树）

A Simple Problem with Integer

题目

您有N个整数，A1, A2, …, AN。您要进行两类操作：一类操作是将某个给定的数在一个给定的区间内加到每个数上；另一类是求以一个给出的区间内的数字的总和。

输入

第一行给出两个数字N和Q，1≤N,Q≤100000;
第二行给出N个数字，是A1, A2, …, AN的初始值，-1000000000≤Ai≤1000000000。
接下来的Q行每行给出一个操作：
“C a b c” 表示将c加到Aa, Aa+1,…, Ab的每个值上面，-10000≤c≤10000；
"Q a b"表示求Aa, Aa+1,…, Ab的总和。

输出

按序对所有的Q条操作给出结果，每条结果一行。

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;
#define ll long long
const int maxn=101000;
int n,m;
ll arr[maxn];
struct node{
	int l,r;
	ll sum,lazy;
	int len;
}tree[maxn<<2];
//更新操作后，该节点还剩的可用空间数量 
void pushup(int rt)
{
	tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum;
}
void mark_down(int rt)
{
	if(tree[rt].lazy)
	{
		tree[rt<<1].sum+=tree[rt].lazy*tree[rt<<1].len;
		tree[rt<<1|1].sum+=tree[rt].lazy*tree[rt<<1|1].len;
		tree[rt<<1].lazy+=tree[rt].lazy;
		tree[rt<<1|1].lazy+=tree[rt].lazy;
		tree[rt].lazy=0;
	}
}
void build(int l,int r,int rt)
{
	tree[rt].l=l;
	tree[rt].r=r;
	tree[rt].lazy=0;
	tree[rt].len=r-l+1;
	if(l==r)
	{
		tree[rt].sum=arr[l];
		return;
	}
	int m=(l+r)/2;
	build(l,m,rt<<1);
	build(m+1,r,rt<<1|1);
	pushup(rt);
}
void update(int data,int l,int r,int rt)
{
	if(tree[rt].l==l&&tree[rt].r==r)
	{
		tree[rt].sum+=data*tree[rt].len;
		tree[rt].lazy+=data;
		return;
	}
	mark_down(rt);
	if(l>tree[rt].r||r<tree[rt].l) return;
	int mid=(tree[rt].l+tree[rt].r)/2;
	if(r<=mid)
		update(data,l,r,rt<<1);
	else if(l>mid)
		update(data,l,r,rt<<1|1);
	else{
		update(data,mid+1,r,rt<<1|1);
		update(data,l,mid,rt<<1);
	}
	pushup(rt);
}
ll query(int l,int r,int rt)
{
	if(tree[rt].l==l&&tree[rt].r==r) return tree[rt].sum;
	mark_down(rt);
	pushup(rt);
	if(r<tree[rt].l||l>tree[rt].r) return 0;
	int mid=(tree[rt].l+tree[rt].r)/2;
	if(r<=mid)
		return query(l,r,rt<<1);
	else if(l>mid)
		return query(l,r,rt<<1|1);
	else{
		ll L=query(l,mid,rt<<1);
		ll R=query(mid+1,r,rt<<1|1);
		return L+R;
	}
}
int main()
{
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		memset(tree,0,sizeof(tree));
		for(int i=1;i<=n;i++)
			scanf("%lld",&arr[i]);
		build(1,n,1);
		while(m--)
		{
			char op[5];
			int a,b;
			scanf("%s%d%d",op,&a,&b);
			if(op[0]=='C')
			{
				ll c;
				scanf("%lld",&c);
				update(c,a,b,1);
			}else{
				printf("%lld\n",query(a,b,1));
			}
		}
	}
	return 0;
}