区间树状数组

来源:互联网 发布:Windows VPN 编辑:程序博客网 时间:2024/05/17 07:07

一开始以为区间树状数组没什么区别,xjb乱搞了后调了好久都调不出来QAQ
首先有一个差分数组di
d[i]=a[i]-a[i-1];
所以a[x]=d[1]+d[2]+…+d[x]
a[x]前缀和:d[1]x+d[2] (x-1)+…+ d[i] * (x-i+1)+…+ d[x] *1
维护 d[i] 和 d[i]*i

i=1xa[i]=(x+1)i=1xd[i]i=1x(d[i]i)

注意:

代码中的e[i]不等于上文的d[i](其实是d[ i ]的树状数组)

f[i]不等于d[i]i(其实是 d[ i ] i 的树状数组)

codevs

Update(10月2日):
这个码风太丑了,以下是我新的区间树状数组的部分代码

void Ins(int x,int y){    for(int i=x;i<=n;i+=lb(i))f[i]+=y,g[i]+=x*y;}LL Sum(int x){    LL sum=0;    for(int i=x;i>=1;i-=lb(i))sum=(sum+(x+1)*f[i]-g[i]+mod)%mod;    return sum;}

完整AC代码(旧)

#include<cstdio>#include<iostream>#define N 200010typedef long long LL;using namespace std;int n,m,o,p,q,x,y;LL e[N],f[N];int lowbit(int x){    return x&(-x);}void update(int x,int y){    int z=y*x;    while(x<=n){        f[x]+=z;        e[x]+=y;        x+=lowbit(x);    }}LL query(int x){    LL sum=0;    while(x>=1){        sum+=e[x];        x-=lowbit(x);    }    return sum;}LL Sum(int x){    LL sum=0;    while(x>=1){        sum+=f[x];        x-=lowbit(x);    }    return sum;}LL Solve(int p,int q){    return (q+1)*query(q)-Sum(q)+Sum(p-1)-p*query(p-1);}int main(){    freopen("data.txt","r",stdin);    scanf("%d",&n);    for(int i=1;i<=n;i++){        scanf("%d",&p);        update(i,p-q);        q=p;    }    scanf("%d",&m);    for(int i=1;i<=m;i++){        scanf("%d",&o);        if(o==1){            scanf("%d%d%d",&p,&q,&y);            update(p,y);            update(q+1,-y);        }        else if(o==2){            scanf("%d%d",&p,&q);            printf("%lld\n",Solve(p,q));        }    }}
原创粉丝点击