首页 > 代码库 > BZOJ 3155: Preprefix sum

BZOJ 3155: Preprefix sum

大意:给一个数组,先求出SUM[I],然后动态的求出1-I的SUM[I]的和,

       这题得化公式:

      树状数组维护两个和:SUM(A[I])(1<=I<=X);

                                  SUM(A[I]*(N-I+1)) (1<=I<=X);

                                 答案就是:SUM(A[I]*(N-I+1))-SUM[A[I]]*(N-X) (1<=I<=X);

                           

#include<stdio.h>#include<string.h>#include<algorithm>#include<cmath>using namespace std;typedef long long ll;const int N=110001;ll s[N],t[N];int a[N];int n;int lowbit(int x){    return x&(-x);}void update1(int x,ll v){    while (x<=n)    {        s[x]+=v;        x+=lowbit(x);    }}void update2(int x,ll v){    while (x<=n)    {        t[x]+=v;        x+=lowbit(x);    }}ll sum1(int x){    ll ans=0;    while (x)    {        ans+=s[x];        x-=lowbit(x);    }    return ans;}ll sum2(int x){    ll ans=0;    while (x)    {        ans+=t[x];        x-=lowbit(x);    }    return ans;}int main(){    int m;    char s[8];    scanf("%d%d",&n,&m);    for (int i=1;i<=n;i++)    {        scanf("%d",&a[i]);        update1(i,a[i]);        update2(i,(ll)a[i]*(n-i+1));    }    while (m--)    {        int x,y;        scanf("%s%d",s,&x);        if (s[0]==Q)        {            printf("%lld\n",sum2(x)-sum1(x)*(n-x));        }        else        {            scanf("%d",&y);            update1(x,y-a[x]);            update2(x,(ll)(y-a[x])*(n-x+1));            a[x]=y;        }    }    return 0;}