HihoCoder 1336 二维树状数组

最新推荐文章于 2019-05-13 21:22:00 发布

数论只会GCD

最新推荐文章于 2019-05-13 21:22:00 发布

阅读量558

点赞数

CC 4.0 BY-SA版权

分类专栏： Hihocoder 数据结构-树状数组文章标签： ACM 树状数组数据结构 HihoCoder

本文链接：https://blog.youkuaiyun.com/ADjky/article/details/53224636

数据结构-树状数组同时被 2 个专栏收录

12 篇文章

订阅专栏

Hihocoder

10 篇文章

订阅专栏

本文详细介绍了二维树状数组的应用及其实现方法，通过解决HihoCoder1336问题来展示如何使用二维树状数组处理矩阵求和，特别关注在求区间和时的划分技巧。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

传送门：HihoCoder 1336

题解

树状数组在二维上的扩展，水题
要注意的是sum的时候要拆分，因为sum是从左上角(1, 1)到(x, y)这个矩阵的sum，所有要进行小划分
树状数组，下标不能为0

AC code:

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;

#define lowbit(x) (x & (-x))
#define LL long long 
#define debug 0

const int maxn(1005);
const int mod(1e9 + 7);

int c[maxn][maxn];

int n, q;



void add(int x, int y, int v) {

    for (int i = x; i <= n; i += lowbit(i)) {
        for (int j = y; j <= n; j += lowbit(j)) {
            c[i][j] += v;
        }
    }
}

LL getSum(int x, int y) {
    LL res = 0;

    for (int i = x; i; i -= lowbit(i)) {
        for (int j = y; j; j -= lowbit(j)) {
            res += c[i][j];
        }
    }

    return res;
}

int main() {
#if debug
    freopen("in.txt", "r", stdin);
#endif //debug

    int x1, y1, x2, y2, v;
    char s[10];

    while (cin >> n >> q) {

        memset(c, 0, sizeof(c));
        //memset(ans, 0, sizeof(ans));

        while(q--) {

            cin >> s;
            if (s[0] == 'A') {
                cin >> x1 >> y1 >> v;
                add(x1 + 1, y1 + 1, v);//传参要注意lowbit边界
            }
            else {
                cin >> x1 >> y1 >> x2 >> y2;
                LL ans = getSum(x2 + 1, y2 + 1) - getSum(x2 + 1, y1) - getSum(x1, y2 + 1) + getSum(x1, y1);//划分sum
                ans %= mod;
                if (ans < 0) ans += mod;//取模处理
                cout << ans << endl;
            }

        }

    }

    return 0;
}