5911: 计数问题（二维树状数组）

5911: 计数问题

时间限制: 1 Sec   内存限制: 128 MB
提交: 187   解决: 53
[ 提交][ 状态][ 讨论版][命题人: admin]

题目描述

 一个n*m的方格，初始时每个格子有一个整数权值。接下来每次有2种操作：
改变一个格子的权值；
求一个子矩阵中某种特定权值出现的个数。

输入

第一行有两个数n,m。
接下来n行，每行m个数，第i+1行第j个数表示格子(i,j)的初始权值。
接下来输入一个整数q。
接下来q行，每行描述一个操作。
操作1：“1 x y c”(不含双引号)。表示将格子(x,y)的权值改成c（1<=x<=n,1<=y<=m,1<=c<=100)。
操作2：“2 x1 x2 y1 y2 c”(不含双引号，x1<=x2,y1<=y2)。表示询问所有满足格子权值为c，且x1<=x<=x2,y1<=y<=y2的格子(x,y)的个数。

输出

对于每个操作2，按照在输入中出现的顺序，依次输出一行一个整数表示所求得的个数。

样例输入

3 3
1 2 3
3 2 1
2 1 3
3
2 1 2 1 2 1
1 2 3 2
2 2 3 2 3 2

样例输出

1
2

提示

对于30%的数据n,m<=30，q<=10000
对于100%的数据 n,m<=300，q<=100000，1<=c<=100

来源

2018山东冬令营 

区间改变，区间统计，容斥定理

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int a[310][310], tree[305][305][105], n, m;  
int lowbit(int x)  {      return x&(-x);  }  
void update(int x, int y, int c, int k)  
{    
    for(int i=x;i<=n;i+=lowbit(i))    
        for(int j=y;j<=m;j+=lowbit(j)) 
            tree[i][j][c]+=k;  
}  
   
int Sum(int x, int y, int c)  
{  
    int res=0;  
    for(int i=x;i>0;i-=lowbit(i))  
        for(int j=y;j>0;j-=lowbit(j))  
            res+=tree[i][j][c];
    return res;  
}  
int main()  
{  
    int sum=0, q, f;  
    scanf("%d %d",&n,&m);  
    for(int i=1;i<=n;i++)  
        for(int j=1;j<=m;j++)  
        {  
            scanf("%d", &a[i][j]);  
            update(i, j, a[i][j], 1);  
        }   
    scanf("%d",&q);  
    while(q--)  
    {   
        scanf("%d",&f);  
        if(f==1)  
        {  
            int x,y,z;  
            scanf("%d %d %d", &x, &y, &z);  
            update(x, y, a[x][y], -1);  
            update(x, y, z, 1);  
            a[x][y]=z;  
        }  
        else 
        {  
            int x1,x2,y1,y2,z;  
            scanf("%d %d %d %d %d", &x1, &x2, &y1, &y2, &z);  
            int sum=Sum(x2, y2, z)-Sum(x2, y1-1, z)-Sum(x1-1, y2, z)+Sum(x1-1, y1-1, z);  
            printf("%d\n",sum);  
        }  
    }  
    return 0;  
}  
/**************************************************************
    Problem: 5911
    User: ldu_reserver201701
    Language: C++
    Result: 正确
    Time:224 ms
    Memory:39620 kb
****************************************************************/