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

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;
}