首页 > 编程语言 > 详细

树状数组

时间:2019-08-17 20:45:59      阅读:78      评论:0      收藏:0      [点我收藏+]

参考: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;
}

 

树状数组

原文:https://www.cnblogs.com/zyddd915/p/11370117.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!