[BZOJ1452][JSOI2009]Count（二维树状数组）

题目描述

传送门

题解

数据范围很小，那么给每一个不同的权值建一个二维树状数组。
其余查询照旧。

代码

#include
#include
#include
using namespace std;

const int max_n=305;
const int max_opt=105;

int n,m,q,x,y,c,x1,y1,x2,y2,opt;
int ans,ans1,ans2,ans3,ans4;
int a[max_n][max_n];
int C[max_opt][max_n][max_n];

inline void add(int loc1,int loc2,int opt,int val){
    for (int i=loc1;i<=n;i+=i&(-i))
      for (int j=loc2;j<=m;j+=j&(-j))
        C[opt][i][j]+=val;
}
inline int query(int loc1,int loc2,int opt){
    int ans=0;
    for (int i=loc1;i>=1;i-=i&(-i))
      for (int j=loc2;j>=1;j-=j&(-j))
        ans+=C[opt][i][j];
    return ans;
}

int main(){
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;++i)
      for (int j=1;j<=m;++j){
        scanf("%d",&x);
        a[i][j]=x;
        add(i,j,a[i][j],1);
      }
    scanf("%d",&q);
    for (int i=1;i<=q;++i){
        scanf("%d",&opt);
        if (opt==1){
            scanf("%d%d%d",&x,&y,&c);
            add(x,y,a[x][y],-1);
            a[x][y]=c;
            add(x,y,a[x][y],1);
        }
        else{
            scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&c);
            ans1=query(x2,y2,c);
            ans2=query(x1-1,y1-1,c);
            ans3=query(x1-1,y2,c);
            ans4=query(x2,y1-1,c);
            ans=ans1-ans3-ans4+ans2;
            printf("%d\n",ans);
        }
    }
}

总结

认真读题！！！