JSOI2009 Count [树状数组]

Posted 2020-10-21 zcdhj

BZOJ

我们可以发现N,M,C都十分的小，那么只要开100个二维BIT来维护就可以了。每种颜色对应一个BIT。

然后查询的时候再用二维前缀和来搞搞就可以了。

开始在BZOJ时候是T了的，后来才发现是BIT开太大了，导致寻址过慢。

#include <bits/stdc++.h>

const int max_n=300+5;

int N,M,Q;
int A[max_n][max_n],bit[101][max_n][max_n];

inline int read()
{
    register int x=0;
    register char ch=getchar();
    while(!isdigit(ch))
        ch=getchar();
    while(isdigit(ch))
    {
        x=(x<<1)+(x<<3)+ch-‘0‘;
        ch=getchar();
    }
    return x;
}

void modify(int c,int x,int y,int num)
{
    for(int i=x;i<=N;i+=i&-i)
        for(int j=y;j<=M;j+=j&-j)
            bit[c][i][j]+=num;
}

int query(int c,int x,int y)
{
    int res=0;
    for(int i=x;i;i-=i&-i)
        for(int j=y;j;j-=j&-j)
            res+=bit[c][i][j];
    return res;
}

int main()
{
    int opt,x,y,xt,yt,c;
    N=read(),M=read();
    for(int i=1;i<=N;++i)
    {
        for(int j=1;j<=M;++j)
        {
            A[i][j]=read();
            modify(A[i][j],i,j,1);
        }
    }
    Q=read();
    while(Q--)
    {
        opt=read();
        if(opt==1)
        {
            x=read(),y=read(),c=read();
            if(A[x][y]==c) continue;
            modify(A[x][y],x,y,-1);
            A[x][y]=c; 
            modify(A[x][y],x,y,1);
        }
        else
        {
           x=read(),xt=read(),y=read(),yt=read(),c=read();
           printf("%d\n",query(c,xt,yt)-query(c,xt,y-1)-query(c,x-1,yt)+query(c,x-1,y-1));
        }
    }
    return 0;
}