如何將普通樹狀數組升級

　　普通的單點修改單點查詢就不講了，從區間修改和單點查詢講起。

　　原來的值存在a[]裏面，多建立個數組c1[],註意：c1[i]=a[i]-a[i-1]。

　　那麽求a[i]的值的時候a[i]=a[i-1]+c1[i]=a[i-2]+c1[i]+c1[i-1]=…..=c1[1]+c1[2]+…+c1[i]。

　　所以就用c1[]建立樹狀數組，便可以很快查詢a[i]的值。不多說，見代碼。

#include<iostream>
#include<cstdio>
#define lb(x) x&-x
#define maxn 1000000
#define
 
 in(x) scanf("%d",&x)
#define in3(x,y,z) scanf("%d%d%d",&x,&y,&z)
using namespace std;
int a[maxn],c1[maxn],n,m,val,x,y,temp;
void update(int x,int val)
{
    while(x<=n)
    {
        c1[x]+=val;
        x+=lb(x);
    }
}
int sum(int x)
{
    int ans=0;
    while(x)
    {
        ans
 
+=c1[x];
        x-=lb(x);
    }
    return ans;
}
main(){
    in(n);
    in(m);
    for(int i=1;i<=n;i++)
    {
        in(a[i]);
        update(i,a[i]-a[i-1]);
    }
    while(m--)
    {
        in(temp);
        if(temp==1)
        {
            in(x);
            printf("%d\n",sum(x));
        }
        
 
else
        {
            in3(x,y,val);
            update(x,val);
            update(y+1,-val);
        }
    }
}

　　自認為還是比較好看懂的，接下來是區間修改和區間查詢了。

　　我們用sum(1,k)表示區間1到k的和。

　　那麽sum(1,k)=c1(1)+(c1(2)+c1(2))+(c1(1)+c1(2)+c1(3))+…+(c1(1)+c1(2)+…+c1(k))。

　　然後我們把式子打開。

　　sum(1,k)=k*(c1(1)+c1(2)+c1(3)+…+c1(k))-(0*c1*(1)+1*c1(2)+2*c1(3)+…+(k-1)*c1(k))。

　　是不是有些小激動，我們可以多建立一個數組c2[]，c2[n]用來存(n-1)*c1(n),並且把c2數組也建立成樹狀數組，那麽問題就迎刃而解了。

　　詳見代碼：

#include<iostream>

#include<cstdio>

#define lb(x) x&-x

#define maxn 1000000

#define in(x) scanf("%d",&x)

#define in3(x,y,z) scanf("%d%d%d",&x,&y,&z)

using namespace std;

int a[maxn],c1[maxn],c2[maxn],n,m,val,x,y,temp;

void update(int *q,int x,int val)

{

         while(x<=n)

         {

                   q[x]+=val;

                   x+=lb(x);

         }

}

int getsum(int *q,int x)

{

         int ans=0;

         while(x)

         {

                   ans+=q[x];

                   x-=lb(x);

         }

         return ans;

}

int sum(int x)

{

         int ans1,ans2;

         ans1=x*getsum(c1,x);

         ans2=getsum(c2,x);

         return ans1-ans2;

}

int inquire(int x,int y)

{

         int ans1,ans2;

         ans1=sum(y);

         ans2=sum(x-1);

         return ans1-ans2;

}

main(){

         in(n);

         in(m);

         for(int i=1;i<=n;i++)

         {

                   in(a[i]);

                   update(c1,i,a[i]-a[i-1]);

                   update(c2,i,(i-1)*(a[i]-a[i-1]));

         }

         for(int i=1;i<=m;i++)

         {

                   in(temp);

                   if(temp==1)

                   {

                            in3(x,y,val);

                            update(c1,x,val);

                            update(c1,y+1,-val);

                            update(c2,x,(x-1)*val);

                            update(c2,y+1,-y*val);

                   }

                   else

                   {

                            in(x);

                            in(y);

                            printf("%d\n",inquire(x,y));

                   }

         }

}

樹狀數組的區間修改與單點查詢與區間查詢