BZOJ3155: Preprefix sum

题解 区间加区间查询线段树裸题  用树状数组实现 常数小

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61

/**************************************************************     Problem: 3155     User: c20161007     Language: C++     Result: Accepted     Time:464 ms     Memory:4416 kb ****************************************************************/    #include <bits/stdc++.h> const int MAXN=1e5+10; #define ll long long using namespace std; ll read(){     ll x=0,f=1;char ch=getchar();     while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}     while(isdigit(ch))x=x*10+ch-'0',ch=getchar();     return f*x; } ll a[MAXN],b[MAXN];int n,q; int get_id(int x){return x&(-x);} void add1(int x,ll vul){     if(x<=0)return ;     for(int i=x;i<=n;i+=get_id(i))a[i]+=vul;     return ; } void add2(int x,ll vul){     if(x<=0)return ;     for(int i=x;i>0;i-=get_id(i))b[i]+=vul; } ll Sum1(int x){     if(x<=0)return 0;     ll ans=0;     for(int i=x;i>0;i-=get_id(i))ans+=a[i];     return ans; } ll Sum2(int x){     if(x<=0)return 0;     ll ans=0;     for(int i=x;i<=n;i+=get_id(i))ans+=b[i];     return ans; } ll A[MAXN],sum[MAXN]; int main(){     n=read();q=read();     for(int i=1;i<=n;i++)A[i]=read();     for(int i=1;i<=n;i++)sum[i]=A[i]+sum[i-1];     int p=n;n++;     for(int i=1;i<=p;i++)add1(i,1ll*sum[i]*i),add2(i,sum[i]),add1(i-1,-1*sum[i]*(i-1)),add2(i-1,-1*sum[i]);     char str[11];int pos;ll vul,key;     for(int i=1;i<=q;i++){     scanf(" %s",str);     if(str[0]=='Q')pos=read(),printf("%lld\n",Sum1(pos)+1ll*pos*Sum2(pos+1));     else{         pos=read();vul=read();key=vul;vul-=A[pos];A[pos]=key;         add1(p,1ll*vul*p),add2(p,vul);         add1(pos-1,-1*vul*(pos-1)),add2(pos-1,-1*vul);     }     }     return 0; }

 　　　　　　　　　　　　　　3155: Preprefix sum
　　　　　　　　　　　　Time Limit: 1 Sec  Memory Limit: 512 MB
　　　　　　　　　　　　Submit: 2060  Solved: 892
　　　　　　　　　　　　[Submit][Status][Discuss]
Description

 

Input

第一行给出两个整数N，M。分别表示序列长度和操作个数
接下来一行有N个数，即给定的序列a1,a2,....an
接下来M行，每行对应一个操作，格式见题目描述

Output

对于每个询问操作，输出一行，表示所询问的SSi的值。

Sample Input
5 3

1 2 3 4 5

Query 5

Modify 3 2

Query 5
Sample Output
35

32
HINT

1<=N,M<=100000,且在任意时刻0<=Ai<=100000

Source

Katharon+#1