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

题目描述

题解

代码

#include<iostream>
#include<cstring>
#include<cstdio>
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);
}
}
}

总结