BZOJ 1452 [JSOI2009]Count

【题目分析】
二维树状数组

---

【代码】

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
int n,m,q;
int t[101][301][301];
int a[301][301];
inline void add(int col,int x,int y,int f)
{
//  cout<<col<<" "<<x<<" "<<y<<" "<<f<<endl;
    for (int i=x;i<=n;i+=i&(-i))
        for (int j=y;j<=m;j+=j&(-j))
            t[col][i][j]+=f;
}
inline int gs(int col,int x,int y)
{
    int ret=0;
    for (int i=x;i;i-=i&(-i))
        for (int j=y;j;j-=j&(-j))
            ret+=t[col][i][j];
    return ret;
}
inline int query(int col,int x1,int x2,int y1,int y2)
{
//  cout<<"query"<<gs(col,x2,y2)<<" "<<gs(col,x1-1,y2)<<" "<<gs(col,x2,y1-1)<<" "<<gs(col,x1-1,y1-1)<<endl;
    return gs(col,x2,y2)-gs(col,x1-1,y2)-gs(col,x2,y1-1)+gs(col,x1-1,y1-1);
}
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(a[i][j],i,j,1);
        }
    scanf("%d",&q);
    for (int i=1;i<=q;++i)
    {
        int opt;
        scanf("%d",&opt);
        if (opt==1){
            int x,y,to;
            scanf("%d%d%d",&x,&y,&to);
            add(a[x][y],x,y,-1);
            a[x][y]=to;
            add(a[x][y],x,y,1);
        }
        else
        {
            int x1,x2,y1,y2,cc;
            scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&cc);
            printf("%d\n",query(cc,x1,x2,y1,y2));
        }
    }
}