bzoj1452: [JSOI2009]Count(二维树状数组)

二维树状数组裸题,开一百个树状数组维护每一个颜色个数即可

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
#define lowbit(x) ((x)&(-x))
const int N=300+10;
int tr[100+10][N][N],a[N][N],n,m,q;
void update(int k,int idx,int idy,int val)
{
    for(int i=idx;i<=n;i+=lowbit(i))
        for(int j=idy;j<=m;j+=lowbit(j))
            tr[k][i][j]+=val;
}
int query(int k,int idx,int idy)
{
    int ret=0;
    for(int i=idx;i;i-=lowbit(i))
        for(int j=idy;j;j-=lowbit(j))
            ret+=tr[k][i][j];
    return ret;
}   
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]);update(a[i][j],i,j,1);}
    scanf("%d",&q);
    for(int op,x1,y1,x2,y2,x,y,c,i=1;i<=q;i++)
    {
        scanf("%d",&op);
        if(op==1)
        {
            scanf("%d%d%d",&x,&y,&c);
            update(a[x][y],x,y,-1);
            update(c,x,y,1);a[x][y]=c;
        }else
        {
            scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&c); 
            printf("%d\n",query(c,x2,y2)-query(c,x2,y1-1)-query(c,x1-1,y2)+query(c,x1-1,y1-1));
        }
    }
    return 0;
}