【SDOI2017】【BZOJ4911】【LOJ2269】切树游戏（FWT）（链分治）（线段树）（矩阵乘法）

LOJ传送门

BZOJ传送门

洛谷上有某位毒瘤上传了专门卡树剖的数据，写完了才看见，以后会重新把全局平衡二叉树的题解写在另一篇博客里面。

---

题解：

首先如果没有修改这道题码长最多也就2K左右。

树上动态DP，喜闻乐见写链分治或者全局平衡二叉树。我写的链分治。

然后就是常见的统计轻儿子，线段树合并重儿子的操作，注意标记的合并，可以去看猫锟的blog

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define gc get_char
#define cs const

namespace IO{
	cs int Rlen=1<<22|1;
	char buf[Rlen],*p1,*p2;	
	inline char get_char(){return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;}
	inline char peek(){return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1;}
	inline char get_alpha(){while(!isalpha(peek()))gc();return gc();}
	
	template<typename T>
	inline T get(){
		char c;
		while(!isdigit(c=gc()));T num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return num;
	}
	inline int getint(){return get<int>();}
}
using namespace IO;

using std::cerr;
using std::cout;

cs int mod=1e4+7;
int inv[mod];
inline int add(int a,int b){return (a+=b)>=mod?a-mod:a;}
inline int dec(int a,int b){return (a-=b)<0?a+mod:a;}
inline int mul(int a,int b){static ll r;r=(ll)a*b;return r>=mod?r%mod:r;}
inline int power(int a,int b,int res=1){
	for(;b;b>>=1,a=mul(a,a))(b&1)&&(res=mul(res,a));
	return res;
}
inline void Inc(int &a,int b){(a+=b)>=mod&&(a-=mod);}
inline void Dec(int &a,int b){(a-=b)<0&&(a+=mod);}
inline void Mul(int &a,int b){a=mul(a,b);}

inline void init_inv(){
	inv[0]=1,inv[1]=1;
	for(int re i=2;i<mod;++i)inv[i]=mul(inv[mod%i],mod-mod/i);
}

struct tag{
	int x,y;
	tag():x(1),y(0){}
	void operator=(int v){v?(x=v,y=0):(x=1,y=1);}
	void operator*=(int v){v?x=mul(x,v):++y;}
	void operator/=(int v){v?x=mul(x,inv[v]):--y;}
	operator int(){return y?0:x;}
};

cs int N=3e4+7,M=129;

int n,S,invS,Q;

struct atom{
	int a[M],b[M],c[M],d[M];
	atom(){}
	void operator=(cs atom &nd){
		memcpy(a,nd.a,sizeof(int)*S);
		memcpy(b,nd.b,sizeof(int)*S);
		memcpy(c,nd.c,sizeof(int)*S);
		memcpy(d,nd.d,sizeof(int)*S);
	}
};

inline void merge(atom &C,cs atom &A,cs atom &B){
	for(int re i=0;i<S;++i){
		C.d[i]=add(add(A.d[i],B.d[i]),mul(B.b[i],A.c[i]));
		C.c[i]=add(mul(B.a[i],A.c[i]),B.c[i]);
		C.b[i]=add(A.b[i],mul(A.a[i],B.b[i]));
		C.a[i]=mul(A.a[i],B.a[i]);
	}
}

inline void FWT(int *A){
	for(int re i=1;i<S;i<<=1)
	for(int re j=0;j<S;j+=i<<1)
	for(int re k=0;k<i;++k){
		int x=A[j|k],y=A[i|j|k];
		A[j|k]=add(x,y),A[i|j|k]=dec(x,y);
	}
}
inline void IFWT(int *A){FWT(A);for(int re i=0;i<S;++i)A[i]=mul(A[i],invS);}

int nxt[N<<1],last[N],to[N<<1],ecnt;
inline void addedge(int u,int v){
	nxt[++ecnt]=last[u],last[u]=ecnt,to[ecnt]=v;
	nxt[++ecnt]=last[v],last[v]=ecnt,to[ecnt]=u;
}

int fa[N],top[N],bot[N],siz[N],son[N];
int in[N],nd[N],clk;

void dfs1(int u,int p){
	fa[u]=p;siz[u]=1;
	for(int re e=last[u],v=to[e];e;v=to[e=nxt[e]])if(v!=p){
		dfs1(v,u);siz[u]+=siz[v];
		if(siz[v]>siz[son[u]])son[u]=v;
	}
}
void dfs2(int u,int p){
	nd[in[u]=++clk]=u;
	if(son[u]){
		top[son[u]]=top[u];
		dfs2(son[u],u);bot[u]=bot[son[u]];
	}
	else {bot[u]=u;return ;}
	for(int re e=last[u],v=to[e];e;v=to[e=nxt[e]])if(v!=p&&v!=son[u])dfs2(top[v]=v,u);
}

int w[M][M],val[N];
tag LF[N][M];
int F[N][M],H[N][M],LH[N][M];

void dfs_dp(int u,int p){
	memcpy(F[u],w[val[u]],sizeof(int)*S);
	for(int re e=last[u],v=to[e];e;v=to[e=nxt[e]])if(v!=p){
		dfs_dp(v,u);
		for(int re i=0;i<S;++i){
			Mul(F[u][i],F[v][i]+1);
			Inc(H[u][i],H[v][i]);
		}
	}
	for(int re i=0;i<S;++i)Inc(H[u][i],F[u][i]);
}

inline void get_L(){
	for(int re u=1;u<=n;++u){
		for(int re i=0;i<S;++i)LF[u][i]=w[0][i];
		for(int re e=last[u],v=to[e];e;v=to[e=nxt[e]])
		if(v!=fa[u]&&v!=son[u]){
			for(int re i=0;i<S;++i){
				LF[u][i]*=add(F[v][i],1);
				Inc(LH[u][i],H[v][i]);
			}
		}
	}
}

atom RES;bool fir;
int *t1=RES.c,*t2=RES.d;
namespace SGT{
#define lc k<<1
#define rc k<<1|1
	atom a[N<<2];
	
	inline void init_node(atom &a,int u){
		int *w=::w[val[u]];
		for(int re i=0;i<S;++i){
			a.a[i]=mul(LF[u][i],w[i]);
			a.d[i]=add(a.a[i],LH[u][i]);
		}
		memcpy(a.b,a.a,sizeof(int)*S);
		memcpy(a.c,a.a,sizeof(int)*S);
	}
	
	inline void build(int k,int l,int r){
		if(l==r)return init_node(a[k],nd[l]);
		int mid=l+r>>1;
		build(lc,l,mid);build(rc,mid+1,r);
		merge(a[k],a[rc],a[lc]);
	}
	
	inline void modify(int k,int l,int r,int p){
		if(l==r)return init_node(a[k],nd[p]);
		int mid=l+r>>1;
		p<=mid?modify(lc,l,mid,p):modify(rc,mid+1,r,p);
		merge(a[k],a[rc],a[lc]);
	}
	
	inline void query(int k,int l,int r,int ql,int qr){
		if(ql<=l&&r<=qr){
			if(fir)RES=a[k],fir=false;
			else merge(RES,RES,a[k]);
			return ;
		}
		int mid=l+r>>1;
		if(qr<=mid)return query(lc,l,mid,ql,qr);
		if(mid<ql)return query(rc,mid+1,r,ql,qr);
		query(rc,mid+1,r,ql,qr);query(lc,l,mid,ql,qr);
	}
#undef lc
#undef rc
}

inline void get_ans(int u){
	fir=true;SGT::query(1,1,n,in[u],in[bot[u]]);
}

inline void modify(){
	int u=getint();
	val[u]=getint();
	while(u){
		int v=fa[top[u]];
		if(v){
			get_ans(top[u]);
			for(int re i=0;i<S;++i){
				LF[v][i]/=add(t1[i],1);
				Dec(LH[v][i],t2[i]);
			}
		}
		SGT::modify(1,1,n,in[u]);
		if(v){
			get_ans(top[u]);
			for(int re i=0;i<S;++i){
				LF[v][i]*=add(t1[i],1);
				Inc(LH[v][i],t2[i]);
			}
		}
		u=v;
	}
}

signed main(){
#ifdef zxyoi
	freopen("cut.in","r",stdin);
#endif
	init_inv();
	n=getint(),S=getint(),invS=inv[S];
	for(int re i=1;i<=n;++i)val[i]=getint();
	for(int re i=1;i<n;++i)addedge(getint(),getint());
	for(int re i=0;i<M;++i)w[i][i]=1,FWT(w[i]);
	dfs1(1,0),dfs2(top[1]=1,0);
	dfs_dp(1,0),get_L();SGT::build(1,1,n);
	Q=getint();
	while(Q--){
		switch(get_alpha()){
			case 'Q':{get_ans(1);IFWT(t2);cout<<t2[getint()]<<"\n";break;}
			case 'C':modify();
		}
	}
	return 0;
}