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

JSOI2009 Count [树状数组]

时间:2018-01-21 19:09:50      阅读:157      评论:0      收藏:0      [点我收藏+]

标签:导致   body   ref   www.   print   its   ==   数组   getc   

BZOJ

我们可以发现N,M,C都十分的小,那么只要开100个二维BIT来维护就可以了。每种颜色对应一个BIT。

然后查询的时候再用二维前缀和来搞搞就可以了。

开始在BZOJ时候是T了的,后来才发现是BIT开太大了,导致寻址过慢。

#include <bits/stdc++.h>

const int max_n=300+5;

int N,M,Q;
int A[max_n][max_n],bit[101][max_n][max_n];

inline int read()
{
    register int x=0;
    register char ch=getchar();
    while(!isdigit(ch))
        ch=getchar();
    while(isdigit(ch))
    {
        x=(x<<1)+(x<<3)+ch-‘0‘;
        ch=getchar();
    }
    return x;
}

void modify(int c,int x,int y,int num)
{
    for(int i=x;i<=N;i+=i&-i)
        for(int j=y;j<=M;j+=j&-j)
            bit[c][i][j]+=num;
}

int query(int c,int x,int y)
{
    int res=0;
    for(int i=x;i;i-=i&-i)
        for(int j=y;j;j-=j&-j)
            res+=bit[c][i][j];
    return res;
}

int main()
{
    int opt,x,y,xt,yt,c;
    N=read(),M=read();
    for(int i=1;i<=N;++i)
    {
        for(int j=1;j<=M;++j)
        {
            A[i][j]=read();
            modify(A[i][j],i,j,1);
        }
    }
    Q=read();
    while(Q--)
    {
        opt=read();
        if(opt==1)
        {
            x=read(),y=read(),c=read();
            if(A[x][y]==c) continue;
            modify(A[x][y],x,y,-1);
            A[x][y]=c; 
            modify(A[x][y],x,y,1);
        }
        else
        {
           x=read(),xt=read(),y=read(),yt=read(),c=read();
           printf("%d\n",query(c,xt,yt)-query(c,xt,y-1)-query(c,x-1,yt)+query(c,x-1,y-1));
        }
    }
    return 0;
}

JSOI2009 Count [树状数组]

标签:导致   body   ref   www.   print   its   ==   数组   getc   

原文地址:https://www.cnblogs.com/zcdhj/p/8324870.html

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