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

题意: 链接

方法: 二维树状数组

解析: 也是好久没写树状数组,从wfy那听来到水题(他说的找他)(然而还是想了想- -毕竟OI)

读完题后的确想用树状数组来做这道题，但是这要统计的是c种颜色，显然需要c个树状数组，还是300x300的。

所以一下子就开始担心起来内存，因为这题是64M的。没算内存前真心不敢码。

300*300*100*4/1024/1024 算起来大概是30多M吧，总之不超，这样才敢打代码.

100个二维树状数组分别维护每种颜色的个数。

#include <stdio.h>
int n , m ;
int map[301][301] ;
int c[101][301][301] ;
int lowbit(int x)
{
    return x&(-x) ;
}
void update(int x , int y , int co , int v)
{
    while(x <= n)
    {
        int t = y ;
        while(t <= m)
        {
            c[co][x][t] += v ;
            t += lowbit(t) ;
        }
        x += lowbit(x) ;
    }
}
int getsum(int x , int y , int co)
{
    int sum = 0 ;
    while(x > 0)
    {
        int t = y ;
        while(t > 0)
        {
            sum += c[co][x][t] ;
            t -= lowbit(t) ;
        }
        x -=lowbit(x) ;
    }
    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" , &map[i][j]) ;
            update(i , j , map[i][j] , 1) ;
        }
    }
    int q ; 
    scanf("%d" , &q) ;
    while(q--)
    {
        int opt ;
        scanf("%d" , &opt) ;
        if(opt == 1)
        {
            int a , b , v ;
            scanf("%d%d%d" , &a , &b , &v) ;
            update(a , b , map[a][b] , -1) ;
            map[a][b] = v ;
            update(a , b , map[a][b] , 1) ;
        }else
        {
            int x1, y1, x2, y2, v;
            scanf("%d%d%d%d%d", &x1, &x2, &y1, &y2, &v);
            int ans = getsum(x2, y2, v) + getsum(x1-1, y1-1, v) - getsum(x2, y1-1, v) - getsum(x1-1, y2, v);
            printf("%d\n", ans);
        }
    }
}