BZOJ1452_Count_KEY

题目传送门

二维树状数组，对于每个颜色开一个树状数组，用容斥求解。

code：

#include <cstdio>
using namespace std;

int read()
{
    char c;while(c=getchar(),c<''||c>'');
    int x=c-'';while(c=getchar(),c>=''&&c<='')x=x*+c-'';
    return x;
}

int N,M,tr[][][],a[][];

void add(int x,int y,int c,int p)
{
    for(int i=x;i<=N;i+=i&-i)
        for(int j=y;j<=M;j+=j&-j)
            tr[c][i][j]+=p;
    return ;
}

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

int main()
{
    N=read(),M=read();
        for(int i=;i<=N;i++)
            for(int j=;j<=M;j++)
                add(i,j,a[i][j]=read(),);
    int Q=read();
        while(Q--){
            int o=read();
            if(o==){
                int x=read(),y=read(),c=read();
                add(x,y,a[x][y],-);
                add(x,y,a[x][y]=c,);
            }
            else{
                int x=read(),fx=read(),y=read(),fy=read(),c=read();
                printf("%d\n",get(fx,fy,c)-get(x-,fy,c)-get(fx,y-,c)+get(x-,y-,c));
            }
        }
}