【BZOJ】1452 [JSOI2009]Count 树套树

题目传送门

这题和上一题的二维树状数组其实差不多。

定义f[k][i][j]表示区间为[i-lowbit(i)+1,i][j-lowbit(j)+1,j]中颜色为k的节点数。

单点修改就把原来该节点颜色用-1更新，现在该节点颜色用1更新。

区间查询也和上一题的更新差不多，差分一下就行了。

（话说这玩意儿是叫差分吧……又被ZZK大佬怼了……难受QwQ）

（好吧这玩意儿不叫差分叫容斥，㕛被ZZK大佬怼了……㕛难受QwQ）

话说这题也可以用线段树套线段树做，但是我觉得没有那个必要吧……

 p.s.orz zzk，是zzk教会了我写输出优化，拿到了BZOJ上的rank23。

附上AC代码：

#include <cstdio>#include <cctype>#define lowbit(x) ((x)&(-x))using namespace std;const int N=310;int n,m,map[N][N],t[110][N][N],q,o,x1,x2,y1,y2,c;inline char nc(void){static char ch[100010],*p1=ch,*p2=ch;return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;}inline void read(int &a){static char c=nc();int f=1;for (;!isdigit(c);c=nc()) if (c=='-') f=-1;for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());a*=f;return;}inline void change(int k,int x,int y,int w){for (int i=x; i<=n; i+=lowbit(i))for (int j=y; j<=n; j+=lowbit(j))t[k][i][j]+=w;return;}inline int query(int k,int x,int y){int sum=0;for (int i=x; i; i-=lowbit(i))for (int j=y; j; j-=lowbit(j))sum+=t[k][i][j];return sum;}inline void print(int x){if (!x) return (void)putchar('0');int a[11],g=0;while (x) a[++g]=x%10,x/=10;while (g) putchar(a[g--]+'0');return;}int main(void){scanf("%d%d",&n,&m);for (int i=1; i<=n; ++i)for (int j=1; j<=m; ++j)read(map[i][j]),change(map[i][j],i,j,1);read(q);while (q--){read(o);switch (o){case 1: read(x1),read(y1),read(c),change(map[x1][y1],x1,y1,-1),change(map[x1][y1]=c,x1,y1,1); break;case 2: read(x1),read(x2),read(y1),read(y2),read(c),print(query(c,x2,y2)-query(c,x1-1,y2)-query(c,x2,y1-1)+query(c,x1-1,y1-1)),putchar('\n'); break;}}return 0;}