bzoj3155Preprefix sum

bzoj3155Preprefix sum

题意：

询问一个数组前缀和数组的前缀和，支持单点修改。

题解：

SSi=sigma(i,1,n)(n-i+1)*ai=(n+1)*Si-sigma(i,1,n)i*ai。然后就只要用树状数组维护ai和i*ai的前缀和就行了。

代码：

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #define inc(i,j,k) for(int i=j;i<=k;i++)
 #define maxn 100010
 #define lb(x) x&-x
 #define ll long long
 using namespace std;

 ll v[maxn][],sm[maxn][]; int n,m; char s[];
 inline int read(){
     char ch=getchar(); int f=,x=;
     while(ch<''||ch>''){if(ch=='-')f=-; ch=getchar();} while(ch>=''&&ch<='')x=x*+ch-'',ch=getchar();
     return f*x;
 }
 void update(int x,ll val,bool b){while(x<=n)sm[x][b]+=val,x+=lb(x);}
 ll query(int x,bool b){ll q=; while(x)q+=sm[x][b],x-=lb(x); return q;}
 int main(){
     n=read(); m=read();
     inc(i,,n)v[i][]=(ll)read(),v[i][]=(ll)i*v[i][],update(i,v[i][],),update(i,v[i][],);
     inc(i,,m){
         scanf("%s",s);
         if(s[]=='Q'){int a=read(); printf("%lld\n",query(a,)*(a+)-query(a,));}
         if(s[]=='M'){
             int a=read(); ll b=(ll)read();
             update(a,b-v[a][],); v[a][]=b; update(a,b*a-v[a][],); v[a][]=b*a;
         }
     }
     return ;
 }

20160630