树状数组

南笙酒味 提交于 2019-11-27 16:25:31

参考:https://hihocoder.com/discuss/question/4956

注意树状数组下标从1开始,所以在主函数运用的时候注意下标的改动

 

二维树状数组的运用:

hiho1336 矩阵 matrix sum

#include <iostream>
#include <cstdio>
#include <cstring>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;

typedef long long LL;

const int maxn=1000+5,
          mod=1e9+7;

LL BIT[maxn][maxn];
int N,m;

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

void add(int x,int y,int val)
{
    for(int i=x;i<maxn;i+=lowbit(i))
    {
        for(int j=y;j<maxn;j+=lowbit(j))
        {
            BIT[i][j]=(BIT[i][j]+val)%mod;
        }
    }
}

int sum(int x,int y)
{
    LL ret=0;
    for(int i=x;i>0;i-=lowbit(i))
    {
        for(int j=y;j>0;j-=lowbit(j))
        {
            ret=(ret+BIT[i][j])%mod;
        }
    }
    return (int)ret;
}

int main()
{
    mem(BIT,0);
    scanf("%d%d",&N,&m);
    char op[10];
    for(int i=0;i<m;i++)
    {
        scanf("%s",op);
        if(op[0]=='A')
        {
            int x,y,val;
            scanf("%d%d%d",&x,&y,&val);
            x++,y++;
            add(x,y,val);
        }
        else
        {
            int x1,y1,x2,y2;
            scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
            x1++,x2++,y1++,y2++;
            int res=(sum(x2,y2)-sum(x1-1,y2)-sum(x2,y1-1)+sum(x1-1,y1-1)+mod)%mod;
            printf("%d\n",res);
        }
    }
    return 0;
}

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!