bzoj 1452: [JSOI2009]Count

2016-06-20

一读完题，就知道这是一个数据结构体，c只有100这么小，那我们就用100课二维树状数组来维护就行了。树状数组绝对是世上最美妙的数据结构(果然我这种蒟蒻只能做这种sb题)

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
#include<algorithm>
#include<queue>
#define M 309
#define ll long long
#define Mo 12345678
using namespace std;
ll read()
{
    char ch=getchar();
    ll x=0,f=1;
    for(;ch<'0'||ch>'9';ch=getchar())
      if(ch=='-')
        f=-1;
    for(;ch>='0'&&ch<='9';ch=getchar())
      x=x*10+ch-'0';
    return x*f;
}
int n,m,a[M][M],shu[109][M][M];
void jia(int x,int i,int j)
{
    for(;i<=n;i+=i&-i)
      for(int k=j;k<=m;k+=k&-k)
        shu[x][i][k]++;
}
void jian(int x,int i,int j)
{
    for(;i<=n;i+=i&-i)
      for(int k=j;k<=m;k+=k&-k)
        shu[x][i][k]--;
}
int query(int x,int i,int j)
{
    int ans=0;
    for(;i;i-=i&-i)
      for(int k=j;k;k-=k&-k)
        ans+=shu[x][i][k];
    return ans;
}
int main()
{
    n=read();
    m=read();
    for(int i=1;i<=n;i++)
      for(int j=1;j<=m;j++)
        {
            a[i][j]=read();
            jia(a[i][j],i,j);
        }
    int q=read();
    for(int i=1;i<=q;i++)
      {
          int k=read(),x1=read(),x2,y1,y2,c;
        if(k==1)
          {
              y1=read();
              c=read();
              jian(a[x1][y1],x1,y1);
              jia(c,x1,y1);
              a[x1][y1]=c;
          }
        else
          {
              x2=read();
              y1=read();
              y2=read();
              c=read();
              printf("%d
",query(c,x2,y2)-query(c,x2,y1-1)-query(c,x1-1,y2)+query(c,x1-1,y1-1));
          }
      }
    return 0;
}