【模板】树状数组

POJ 3468

题意：长度为n的序列，q次操作，C l，r，x表示区间 l~r 增加x，Q l，r表示求区间 l~r 的和。

#include
#include
#include
using namespace std;
const long long maxn=1e6+10;
long long a[maxn],sum[maxn];
long long lowbit(long long x)
{
    return x&(-x);
}
long long n,q;
void add(long long p,long long x)
{
    for(long long i=p;i<=n;i+=lowbit(i))
    {
        a[i]+=x;
        sum[i]+=p*x;
    }
}
long long query(long long p)
{
    long long ans=0;
    for(long long i=p;i>0;i-=lowbit(i))
    {
        ans+=(p+1)*a[i]-sum[i];
    }
    return ans;
}
char s[10];
int main()
{

    scanf("%lld%lld",&n,&q);
    for(long long i=1;i<=n;i++)
    {
        long long x;
        scanf("%lld",&x);
        add(i,x);
        add(i+1,-x);
    }
    for(long long i=1;i<=q;i++)
    {
        scanf("%s",s);

        if(s[0]=='C')
        {
            long long l,r,x;
            scanf("%lld%lld%lld",&l,&r,&x);
            add(l,x);
            add(r+1,-x);
        }
        if(s[0]=='Q')
        {
            long long l,r;
            scanf("%lld%lld",&l,&r);
            long long ans=query(r)-query(l-1);
            printf("%lld\n",ans);
        }
    }
}

二维树状数组

题意：C x1，y1，x2，y2，k表示矩阵加k，Q x1，y1，x2，y2表示询问矩阵和。

#include
#include
#include
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=2005;
struct bit
{
    int a[N][N];
    inline int lowbit(int x)
    {
        return x&(-x);
    }
    inline void add(int x,int y,int t)
    {
        int i,j;
        for(i=x;i0;i-=lowbit(i))
        {
            for(j=y;j>0;j-=lowbit(j))ans+=a[i][j];
        }
        return ans;
    }
}a,b,c,d;


inline void add(int x1,int y1,int x2,int y2,int t)
{
    a.add(x1,y1,t),a.add(x1,y2+1,-t);
    a.add(x2+1,y1,-t),a.add(x2+1,y2+1,t);

    b.add(x1,y1,t*x1); b.add(x2+1,y1,-t*(x2+1));
    b.add(x1,y2+1,-t*x1); b.add(x2+1,y2+1,t*(x2+1));

    c.add(x1,y1,t*y1); c.add(x2+1,y1,-t*y1);
    c.add(x1,y2+1,-t*(y2+1)); c.add(x2+1,y2+1,t*(y2+1));

    d.add(x1,y1,t*x1*y1); d.add(x2+1,y1,-t*(x2+1)*y1);
    d.add(x1,y2+1,-t*x1*(y2+1)); d.add(x2+1,y2+1,t*(x2+1)*(y2+1));
}
inline int get(int x,int y)
{
    return a.get(x,y)*(x+1)*(y+1)-b.get(x,y)*(y+1)-(x+1)*c.get(x,y)+d.get(x,y);
}
inline int get(int x1,int y1,int x2,int y2)
{
    return get(x2,y2)-get(x2,y1-1)-get(x1-1,y2)+get(x1-1,y1-1);
}
int main()
{
    char op[2];
    while (~scanf("%s",op))
    {
        int x1,y1,x2,y2,k;
        if (op[0]=='C')
        {
            scanf("%d%d",&x1,&y1);
            scanf("%d%d%d",&x2,&y2,&k);
            add(x1,y1,x2,y2,k);
        }
        if(op[0]=='Q')
        {
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            printf("%d\n",get(x1,y1,x2,y2));
        }
    }
}