Time Limit: 10 Sec
Memory Limit: 64 MB
Description
一个N*M的方格,初始时每个格子有一个整数权值,接下来每次有2个操作:
改变一个格子的权值
求一个子矩阵中某个特定权值出现的个数
Input
每一行有两个数字N,M
接下来N行,每行M个数字。第i+1行第j个数字表示格子(i,j)的初值
接下来输入一个Q,后面Q行每行描述一个操作
操作1:
1 x y c,表示将格子(x,y)的值变为c
操作2:
2 x1 x2 y1 y2 c,表示询问所有满足格子中数字为c的格子数字
(n,m<=300,Q<=5000)
(1<=x<=N,1<=y<=M,1<=c<=100)
(x1<=x<=x2,y1<=y<=y2)
Output
对于每个操作2,按输入中出现的顺序,依次输出一行一个整数表示所求得的个数
题目分析
第一眼觉得很难,再一看数据范围
颜色数字这么小直接对每个颜色建一个二维树状数组就好了啊
连离散化都不用
二维树状数组其实也就是多套了一层循环而已
#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
#define lowbit(x) ((x)&(-x))
int read()
{
int f=1,x=0;
char ss=getchar();
while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
return f*x;
}
int n,m,q;
int sum[110][310][310];//第一维表示的是颜色
int color[310][310];
void add(int col,int x,int y,int v)
{
for(int i=x;i<=n;i+=lowbit(i))
for(int j=y;j<=m;j+=lowbit(j))
sum[col][i][j]+=v;
}
int get(int col,int x,int y)
{
int ans=0;
for(int i=x;i>0;i-=lowbit(i))
for(int j=y;j>0;j-=lowbit(j))
ans+=sum[col][i][j];
return ans;
}
int main()
{
n=read();m=read();
for(int i=1;i<=n;++i)
for(int j=1;j<=m;++j)
color[i][j]=read(),add(color[i][j],i,j,1);
q=read();
while(q--)
{
int k=read();
if(k==1)
{
int x=read(),y=read(),col=read();
add(color[x][y],x,y,-1);
color[x][y]=col;
add(color[x][y],x,y,1);
}
else
{
int x1=read(),x2=read(),y1=read(),y2=read(),col=read();
int ans=get(col,x2,y2)-get(col,x2,y1-1)-get(col,x1-1,y2)+get(col,x1-1,y1-1);
printf("%d\n",ans);
}
}
return 0;
}