bzoj1452: [JSOI2009]Count(树状数组)

1503: [NOI2004]郁闷的出纳员

题目：传送门 

 

---

 

 

题解：

　　 二维树状数组裸题

　　 将a[x][y][c]表示的是[1,1]到[x,y]这个矩阵中值为c的格子个数（二维前缀和思想）

　　 最后利用矩阵区间求和的方法加加减减就OK

 　　 

---

 

 

代码：

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
int n,m;
int mp[310][310],a[310][310][110];
int lowbit(int x){return x&-x;}
void change(int x,int y,int c,int d)
{
    int t=y;
    while(x<=n)
    {
        y=t;
        while(y<=m)
        {
            a[x][y][c]+=d;
            y+=lowbit(y);
        }
        x+=lowbit(x);
    }
}
int getsum(int x,int y,int c)
{
    int t=y,sum=0;
    while(x)
    {
        y=t;
        while(y)
        {
            sum+=a[x][y][c];
            y-=lowbit(y);
        }
        x-=lowbit(x);
    }
    return sum;
}
int main()
{
    scanf("%d%d",&n,&m);memset(a,0,sizeof(a));
    for(int i=1;i<=n;i++)for(int j=1;j<=m;j++)scanf("%d",&mp[i][j]),change(i,j,mp[i][j],1);
    int Q;scanf("%d",&Q);
    while(Q--)
    {
        int opt;scanf("%d",&opt);
        if(opt==1)
        {
            int x,y,c;scanf("%d%d%d",&x,&y,&c);
            change(x,y,mp[x][y],-1);mp[x][y]=c;change(x,y,c,1);
        }
        else
        {
            int x1,x2,y1,y2,c;scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&c);
            printf("%d\n",getsum(x2,y2,c)-getsum(x1-1,y2,c)-getsum(x2,y1-1,c)+getsum(x1-1,y1-1,c));
        }
    }
    return 0;
}