hihocoder #1336 : Matrix Sum 二维线段树 或 二维树状数组模板

题目链接
解法一：
二维树状数组

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
using namespace std;
typedef long long ll;
const double PI = acos(-1.0);
const double eps = 1e-6;
const int INF = 1000000000;
const int maxn = 1234;
const int mod = 1e9+7;
int T,n,m;
char op[4];
int sx,sy,ex,ey;
int c[maxn][maxn];

int lowbit(int x)
{
    return x&(-x);
}

ll query_sum(int x,int y)
{
    ll sum=0;
    for(int i=x;i!=0;i-=lowbit(i))
    {
        for(int j=y;j!=0;j-=lowbit(j))
        {
            sum=(sum+c[i][j])%mod;
        }
    }
    return sum;
}

void update(int x,int y,int v)
{
    for(int i=x;i<maxn;i+=lowbit(i))
    {
        for(int j=y;j<maxn;j+=lowbit(j))
        {
            c[i][j]=(c[i][j]+v)%mod;
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    {
        memset(c,0,sizeof(c));
        int a,b,w;
        while(m--)
        {
            scanf("%s",op);
            if(op[0]=='A')
            {
                scanf("%d%d%d",&a,&b,&w);
                a++,b++;
                update(a,b,w);
            }
            else
            {
                scanf("%d%d%d%d",&sx,&sy,&ex,&ey);
                sx++,sy++,ex++,ey++;
                printf("%lld\n",(query_sum(ex,ey)-query_sum(sx-1,ey)-query_sum(ex,sy-1)+query_sum(sx-1,sy-1)+mod)%mod);
            }
        }
    }
    return 0;
}

解法二：
二维线段树

#include<bits/stdc++.h>
using namespace std;
const int MOD=1e9+7;
long long sum[4001][4001];
int n;
void cloadd(int tag,int p,int k,int l,int r,int y,long long val)
{
    if(l==r)
    {
        if(tag)sum[p][k]=(sum[p][k]+val)%MOD;
        else sum[p][k]=(sum[2*p][k]+sum[2*p+1][k])%MOD;
        return;
    }
    if(y<=(l+r)/2)cloadd(tag,p,2*k,l,(l+r)/2,y,val);
    else cloadd(tag,p,2*k+1,(l+r)/2+1,r,y,val);
    if(tag)sum[p][k]=(sum[p][2*k]+sum[p][2*k+1])%MOD;
    else sum[p][k]=(sum[2*p][k]+sum[2*p+1][k])%MOD;
}
void rowadd(int k,int l,int r,int x,int y,long long val)
{
    if(x==l&&x==r){cloadd(1,k,1,1,n,y,val);return;}
    if(x<=(l+r)/2)rowadd(2*k,l,(l+r)/2,x,y,val);
    else rowadd(2*k+1,(l+r)/2+1,r,x,y,val);
    cloadd(0,k,1,1,n,y,val);
}
long long cloask(int p,int k,int l,int r,int y,int y1)
{
    if(y<=l&&y1>=r)return sum[p][k]%MOD;
    if(y1<=(l+r)/2)return cloask(p,2*k,l,(l+r)/2,y,y1)%MOD;
    else if(y>=(l+r)/2+1)return cloask(p,2*k+1,(l+r)/2+1,r,y,y1)%MOD;
    return (cloask(p,2*k,l,(l+r)/2,y,(l+r)/2)+cloask(p,2*k+1,(l+r)/2+1,r,(l+r)/2+1,y1))%MOD;
}
long long rowask(int k,int l,int r,int x,int y,int x1,int y1)
{
    if(x==l&&x1==r)return cloask(k,1,1,n,y,y1)%MOD;
    if(x1<=(l+r)/2)return rowask(2*k,l,(l+r)/2,x,y,x1,y1)%MOD;
    else if(x>=(l+r)/2+1)return rowask(2*k+1,(l+r)/2+1,r,x,y,x1,y1)%MOD;
    return (rowask(2*k,l,(l+r)/2,x,y,(l+r)/2,y1)+rowask(2*k+1,(l+r)/2+1,r,(l+r)/2+1,y,x1,y1))%MOD;
}
int main()
{
    int T;
    scanf("%d%d",&n,&T);
    memset(sum,0,sizeof sum);
    while(T--)
    {
        char op[10];
        int x,y,x1,y1;
        long long val;
        scanf("%s",op);
        if(strcmp(op,"Add")==0)
        {
            scanf("%d%d%lld",&x,&y,&val);
            rowadd(1,1,n,x+1,y+1,val);
        }
        else
        {
            scanf("%d%d%d%d",&x,&y,&x1,&y1);
            printf("%lld\n",(rowask(1,1,n,x+1,y+1,x1+1,y1+1)%MOD+MOD)%MOD);//WA的注意了，输出非负整数，先加上1e9+7再取模
        }
    }
    return 0;
}