【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;
}