BZOJ1452 || 洛谷P4054 [JSOI2009]计数问题【二维树状数组】

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;
}