【洛谷P4054】计数问题

题目大意：维护 N*M 个点，每个点有三个权值，支持单点修改，查询矩形区间内权值等于某个值的点的个数。

题解：矩阵可以看成两个维度，权值为第三个维度，为一个三维偏序维护问题。发现第三维仅仅为单点修改和单点询问，直接用数组实现最简单，且空间足够。因此，直接建立 100 个二维树状数组，转变成为单点修改，矩形查询的问题。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=301;

int n,m,q;
int t[101][maxn][maxn],d[maxn][maxn];
inline int lowbit(int x){return x&-x;}
inline void modify(int o,int x,int y,int val){
    for(int i=x;i<=n;i+=lowbit(i))
        for(int j=y;j<=m;j+=lowbit(j))
            t[o][i][j]+=val;
}
inline int query(int o,int x,int y){
    int res=0;
    for(int i=x;i;i-=lowbit(i))
        for(int j=y;j;j-=lowbit(j))
            res+=t[o][i][j];
    return res;
}

void read_and_parse(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++){
            scanf("%d",&d[i][j]);
            modify(d[i][j],i,j,1);
        }
}

void solve(){
    scanf("%d",&q);
    while(q--){
        int opt;
        scanf("%d",&opt);
        if(opt==1){
            int x,y,c;scanf("%d%d%d",&x,&y,&c);
            modify(d[x][y],x,y,-1),modify(d[x][y]=c,x,y,1);
        }else{
            int x1,y1,x2,y2,c;scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&c);
            printf("%d\n",query(c,x2,y2)-query(c,x1-1,y2)-query(c,x2,y1-1)+query(c,x1-1,y1-1));
        }
    }
}

int main(){
    read_and_parse();
    solve();
    return 0;   
}