[BZOJ1452][JSOI2009]Count

原题地址

二维树状数组.

没看清数据范围MLE了一发,下次一定要注意看数据范围…

AC code:

#include <cstdio>
const int N=310;
int n,m,q;
int a[N][N];
int T[N][N][101];

void add(int x,int y,int v,int c){
    for(int i=x;i<=n;i+=i&-i)
        for(int j=y;j<=m;j+=j&-j)
            T[i][j][c]+=v;
}

int get(int x,int y,int c){
    int sum=0;
    for(int i=x;i;i-=i&-i)
        for(int j=y;j;j-=j&-j)
            sum+=T[i][j][c];
    return sum;
}

int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            scanf("%d",&a[i][j]);
            add(i,j,1,a[i][j]);
        }
    }
    scanf("%d",&q);
    for(int i=1;i<=q;i++){
        int t,x1,x2,y1,y2,c;
        scanf("%d",&t);
        if(t==1){
            scanf("%d%d%d",&x1,&y1,&c);
            add(x1,y1,-1,a[x1][y1]);
            a[x1][y1]=c;
            add(x1,y1,1,c);
        }
        else{
            scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&c);
            printf("%d\n",get(x2,y2,c)-get(x1-1,y2,c)-get(x2,y1-1,c)+get(x1-1,y1-1,c));
        }
    }

    return 0;
}