每日一题- P4054

怎么是树状数组板子,这也蓝

#include<bits/stdc++.h>
using namespace std;
int n,m,q,a[305][305];
int c[305][305][105];
void add(int x,int y,int k,int col){
	for(int i=x;i<=n;i+=i&-i)
		for(int j=y;j<=m;j+=j&-j)
			c[i][j][col]+=k;
}
int query(int x,int y,int col){
	int res=0;
	for(int i=x;i;i-=i&-i)
		for(int j=y;j;j-=j&-j)
			res+=c[i][j][col];
	return res;
}
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(i,j,1,a[i][j]);
		}
	int q;scanf("%d",&q);
	while(q--){
		int op;scanf("%d",&op);
		if(op==1){
			int x,y,col;scanf("%d%d%d",&x,&y,&col);
			add(x,y,-1,a[x][y]);
			a[x][y]=col;
			add(x,y,1,a[x][y]);
		}else{
			int x1,y1,x2,y2,col;scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&col);
			printf("%d\n",query(x2,y2,col)-query(x1-1,y2,col)-query(x2,y1-1,col)+query(x1-1,y1-1,col));
		}
	}
	return 0;
}
posted @ 2024-07-27 17:13  Kent530  阅读(31)  评论(0)    收藏  举报