1452: [JSOI2009]Count

Time Limit: 10 Sec  Memory Limit: 64 MB
Submit: 3135  Solved: 1828
[Submit][Status][Discuss]

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,按输入中出现的顺序，依次输出一行一个整数表示所求得的个数

Sample Input

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

Sample Output

1
2

 

逛了一天的漫展，好累Orz

重新学习一下树状数组（我什么时候学过来着

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

int n,m,q;
int mp[305][305];
int t[105][305][305];

int lowbit(int x){return x&(-x);}
void update(int x,int y,int c,int v)
{
    for(int i=x;i<=n;i+=lowbit(i))
        for(int j=y;j<=m;j+=lowbit(j))
            t[c][i][j]+=v;
}

int ask(int x,int y,int c)
{
    int tmp=0;
    for(int i=x;i;i-=lowbit(i))
        for(int j=y;j;j-=lowbit(j))
            tmp+=t[c][i][j];
    return tmp;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        {
            scanf("%d",&mp[i][j]);
            update(i,j,mp[i][j],1);
        }
    scanf("%d",&q);
    while(q--)
    {
        int f;
        scanf("%d",&f);
        int x1,y1,x2,y2,c;
        if(f==1)
        {
            scanf("%d%d%d",&x1,&y1,&c);
            update(x1,y1,mp[x1][y1],-1);
            mp[x1][y1]=c;
            update(x1,y1,mp[x1][y1],1);
        }
        else
        {
            scanf("%d%d%d%d%d",&x1,&x2,&y1,&y2,&c);
            int ans=ask(x1-1,y1-1,c)+ask(x2,y2,c)-ask(x1-1,y2,c)-ask(x2,y1-1,c);
            printf("%d\n",ans);
        }
    }
    return 0;
}