BZOJ 1452: [JSOI2009]Count 二维树状数组

最新推荐文章于 2019-04-22 19:24:12 发布

原创最新推荐文章于 2019-04-22 19:24:12 发布 · 421 阅读

0 ·

CC 4.0 BY-SA版权

ACM/ICPC_ BZOJ 专栏收录该内容

222 篇文章

订阅专栏

本文介绍了一种使用二维树状数组解决特定类型问题的方法。通过动态更新和查询操作实现区间统计，适用于需要高效处理二维矩阵中元素变化及区域求和的场景。

这里写图片描述

Sample Output
1

2
这里写图片描述

解法：二维树状数组。数组开到快1000w竟然没炸。

///BZOJ 1452

#include <bits/stdc++.h>
using namespace std;
int n, m, q;
int mp[305][305];
int tr[105][305][305];
inline int lowbit(int x){return x&-x;}
inline void update(int x, int y, int c, int val){
    for(int i=x; i<=n; i+=lowbit(i))
        for(int j=y; j<=n; j+=lowbit(j))
            tr[c][i][j]+=val;
}
inline int ask(int x, int y, int c){
    int tmp = 0;
    for(int i=x; i; i-=lowbit(i)){
        for(int j=y; j; j-=lowbit(j)){
            tmp+=tr[c][i][j];
        }
    }
    return tmp;
}
int main()
{
    scanf("%d%d", &n,&m);
    for(int i=1; i<=n; i++){
        for(int j=1; j<=m; j++){
            scanf("%d", &mp[i][j]);
            update(i,j,mp[i][j],1);
        }
    }
    scanf("%d", &q);
    while(q--){
        int op,x1,x2,y1,y2,c;
        scanf("%d", &op);
        if(op==1){
            scanf("%d%d%d",&x1,&y1,&c);
            update(x1,y1,mp[x1][y1],-1);
            mp[x1][y1]=c;
            update(x1,y1,mp[x1][y1],1);
        }
        else{
            scanf("%d%d%d%d%d", &x1,&x2,&y1,&y2,&c);
            int ans=0;
            ans=ask(x2,y2,c)+ask(x1-1,y1-1,c);
            ans-=ask(x1-1,y2,c);
            ans-=ask(x2,y1-1,c);
            printf("%d\n", ans);
        }
    }
    return 0;
}