zufe 神奇的序列计算 (树状数组)

树状数组维护一下增加的值！！好题啊

Description

小胖是一个数学爱好者，他总是喜欢提出一些奇怪的问题，给出一个长度为N的序列，每次提出l,r（1<=l<=r<=N）,要求求出
a[l]*1+a[l+1]*2+...+a[r]*(r-l+1)的值(序列中数字均大于等于0)，其中1<=N<=100000,询问共M次，1<=M<=100000，询问操作有两种，第一种为更新序列，
形式如：add X 表示a[x]的值增加1，第二种为query L R  表示询问L，R区间的值，求和方式如上面所描述的(所有输入都在整数范围内)。

Input

第一行输入N，M，N为序列大小，M为询问个数。接下来一行为N个数字，表示初始序列。接下来M行，每行开头为一个字符串，如果字符串为
add，则为增加操作，那么之后有一个数字X，表示要增加1的序列下标。如果为query,后面有两个数字 L，R，表示要询问L，R区间的和。
这里由于答案会比较大，我们要求输出答案对1000000007取余。

Output

对于query的询问，输出一个数字，表示得到的答案对1000000007取余的结果。

Sample Input

5 4
1 2 3 4 5
query 1 2
query 3 4
add 1
query 1 2

Sample Output

5
11
6

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const LL maxn=100000+100;
const LL mod=1000000007;
char str[10];
LL sum[maxn],bit1[maxn],bit2[maxn],n,a[maxn],pre[maxn]; // sum系数和， pre 求1的和 
LL sum1(int i) // 求1的和 
{
	LL s=0;
	while(i>0) {
		s=(s+bit1[i])%mod;
		i-=i&(-i);
	} 
	return s;
}
void add1(int i,LL x)
{
	while(i<=n) {
		bit1[i]=(bit1[i]+x)%mod;
		i+=i&(-i);
	}
}

LL sum2(int i)   //求系数之和 
{
	LL s=0;
	while(i>0) {
		s=(s+bit2[i])%mod;
		i-=i&(-i);
	} 
	return s;
}
void add2(int i,LL x)
{
	while(i<=n) {
		bit2[i]=(bit2[i]+x)%mod;
		i+=i&(-i);
	}
}

int main()
{
	int m,t;
	LL l,r,t1,t2,ans,i,j;
	while(scanf("%lld%d",&n,&m)!=EOF) {
		memset(sum,0,sizeof(sum));
		memset(bit1,0,sizeof(bit1));
		memset(bit2,0,sizeof(bit2));
		memset(pre,0,sizeof(pre));

		for(i=1;i<=n;i++) {
			scanf("%lld",&a[i]);
			a[i]=a[i]%mod;
			sum[i]=(sum[i-1]+a[i]*i%mod)%mod;
			pre[i]=(pre[i-1]+a[i])%mod;
		}
		while(m--) {
			scanf("%s",str);
			if(strcmp(str,"query")==0) {
				scanf("%lld%lld",&l,&r);
				t1=(sum[r]-sum[l-1]+mod-(pre[r]-pre[l-1])*(l-1)%mod+mod)%mod;
				t2=(sum2(r)-sum2(l-1)+mod-(sum1(r)-sum1(l-1))*(l-1)%mod+mod)%mod;
				ans=(t1+t2)%mod;
				printf("%lld\n",ans);
			}
			else {
				scanf("%d",&t);
				add1(t,LL(1));
				add2(t,LL(t));
			}
		}
	}
	return 0;
}