码迷,mamicode.com
首页 > 编程语言 > 详细

[二维树状数组]计数问题

时间:2018-05-27 13:35:51      阅读:178      评论:0      收藏:0      [点我收藏+]

标签:代码   结果   sample   fine   bit   str   style   span   amp   

题目描述

 一个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

思路:求询问区间内某一个值出现的次数;
单点更新:若在某个位置(i,j)出现了某个数k则标记为1(三位数组c[i][j][k]记录这个信息),原先的数标记为0;
区间求和:求区间内某个数k出现次数即调用getsum(i,j,k)求和c[..][..][k]数组得到(1,1)-(i,j)内k出现的次数;结果是getsum(xx,yy,k)-getsum(xx,y-1,k)-getsum(x-1,yy,k)+getsum(x-1,y-1,k);(容斥原理)
AC代码:
#include <iostream>
#include<cstdio>
#define lowbit(x) x&(-x)
using namespace std;

int n,m;
int Map[310][310];
int c[310][310][110];

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

int getsum(int x,int y,int k){
   int ret=0;
   for(int i=x;i>0;i-=lowbit(i)){
     for(int j=y;j>0;j-=lowbit(j)){
        ret+=c[i][j][k];
     }
   }
   return ret;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            scanf("%d",&Map[i][j]);
            add(i,j,Map[i][j],1);
        }
    }
    int q;
    scanf("%d",&q);
    while(q--){
        int op;
        scanf("%d",&op);
        if(op==1){
            int x,y,k;
            scanf("%d%d%d",&x,&y,&k);
            add(x,y,Map[x][y],-1);
            add(x,y,k,1);
            Map[x][y]=k;
        }
        if(op==2){
            int x,xx,y,yy,k;
            scanf("%d%d%d%d%d",&x,&xx,&y,&yy,&k);
            printf("%d\n",getsum(xx,yy,k)-getsum(x-1,yy,k)-getsum(xx,y-1,k)+getsum(x-1,y-1,k));
        }
    }
    return 0;
}
 

[二维树状数组]计数问题

标签:代码   结果   sample   fine   bit   str   style   span   amp   

原文地址:https://www.cnblogs.com/lllxq/p/9095555.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!