[bzoj 1452] Count (树状数组)

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,m,q;
int a[305][305];
int t[105][305][305];//每一个权值建立的二维数组
int lowbit(int x){
return x&(-x);
}
void update(int x,int y,int p,int k){
for(int i=x;i<=n;i+=lowbit(i))
for(int j=y;j<=m;j+=lowbit(j))
t[p][i][j]+=k;
}
int getsum(int x,int y,int p){
int sum=0;
for(int i=x;i;i-=lowbit(i))
for(int j=y;j;j-=lowbit(j))
sum+=t[p][i][j];
return sum;
}
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]);
update(i,j,a[i][j],1);
}
}
scanf("%d",&q);
for(int i=1;i<=q;i++){
int k;
scanf("%d",&k);
int x1,x2,y1,y2,c;
if(k==1){
scanf("%d %d %d",&x1,&y1,&c);
update(x1,y1,a[x1][y1],-1);
a[x1][y1]=c;
update(x1,y1,a[x1][y1],1);
}
else{
scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&c);
int ans=0;
ans=getsum(x2,y2,c)-getsum(x1-1,y2,c)-getsum(x2,y1-1,c)+getsum(x1-1,y1-1,c);
cout<<ans<<endl;
}
}
return 0;
}