冬令营--（二维树状数组）

5911: 计数问题

时间限制: 1 Sec   内存限制: 128 MB
提交: 185   解决: 51
[ 提交][ 状态][ 讨论版][命题人: 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

题解：为了计算区间中某元素的个数和区间中单点的修改，可以使用二维树状数组。与一维的树状数组相比较而言，添加和修改各增加了一个for循环，同时进行区间元素查询时运用一点容斥原理，query(x,y,xx,yy,c)=ask(x,y,c)-ask(x-1,y,c)-ask(x,y-1,c)+ask(x-1,y-1,c);

#include<stdio.h>
#include <algorithm>
#include<iostream>
#include<string.h>
#include<vector>
#include<stdlib.h>
#include<math.h>
#include<queue>
#include<deque>
#include<ctype.h>
#include<map>
#include<set>
#include<stack>
#include<string>
#include<algorithm>
#define ll long long
#define INF 0x3f3f3f3f
#define FAST_IO ios::sync_with_stdio(false)
const double PI = acos(-1.0);
const double eps = 1e-6;
using namespace std;

int a[405][405],sum[305][305][205],n,m;

int lowbit(int x){return x&(-x);}

void add(int x,int y,int c,int d)
{
    for(int i=x;i<=n;i+=lowbit(i))
        for(int j=y;j<=m;j+=lowbit(j))
            sum[i][j][c]+=d;
}

int ask(int x,int y,int c)
{
    int ans=0;

    for(int i=x;i>0;i-=lowbit(i))
        for(int j=y;j>0;j-=lowbit(j))
            ans+=sum[i][j][c];

    return ans;
}

int query(int x1,int y1,int x2,int y2,int c)
{
    return ask(x2,y2,c)-ask(x1-1,y2,c)-ask(x2,y1-1,c)+ask(x1-1,y1-1,c);
}

int main()
{
    cin>>n>>m;

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

    int Q;
    cin>>Q;
    while(Q--)
    {
        int node;
        scanf("%d",&node);
        if(node==1)
        {
            int x,y,c;
            scanf("%d%d%d",&x,&y,&c);
            add(x,y,a[x][y],-1);
            a[x][y]=c;
            add(x,y,c,1);
        }
        else
        {
            int x,y,xx,yy,c;
            scanf("%d%d%d%d%d",&x,&xx,&y,&yy,&c);
            printf("%d\n",query(x,y,xx,yy,c));
        }
    }
    return 0;
}