CF1498F Christmas Game

从简单往复杂推。

菊花图并且 \(k=1\) 时，这是个 \(Nim\) 游戏。

继续 \(k=1\)，奇偶分类后发现一样是奇数深度意义下的 \(Nim\) 游戏。

\(k > 1\) 时，深度除以 \(k\) 后还是个 \(Nim\) 游戏。

换根 \(dp\) 求解一下即可。

#include<cstdio> 
const int H=4000040;
char G[H],*_=G;
int read(){
	int x=0;
	while(*_<'0'||'9'<*_)_++;
	while('0'<=*_&&*_<='9')x=x*10+(*_++^'0');
	return x;
}
const int N=100010,K=20;
struct edge{
	edge*net;int to;
}e[N<<1],*tot=e,*head[N];
int n,k,f[N][K<<1],g[K<<1],ans[N];
void dfs1(int u,int fa){
	for(edge*i=head[u];i;i=i->net)if(i->to^fa){
		dfs1(i->to,u);
		for(int j=0;j<k;j++)f[u][(j+1)%k]^=f[i->to][j];
	}
}
void dfs2(int u,int fa){
	int j;
	for(j=k>>1;j<k;j++)ans[u]^=f[u][j];
	for(edge*i=head[u];i;i=i->net)if(i->to^fa){
		for(j=0;j<k;j++)g[(j+1)%k]=f[u][(j+1)%k]^f[i->to][j];
		for(j=0;j<k;j++)f[i->to][(j+1)%k]^=g[j];
		dfs2(i->to,u);
	}
}
int main(){
	fread(G,1,H,stdin);
	int i,u,v;
	n=read(),k=read(),k<<=1;
	for(i=1;i<n;i++)u=read(),v=read(),*++tot={head[u],v},head[u]=tot,*++tot={head[v],u},head[v]=tot;
	for(i=1;i<=n;i++)f[i][0]=read();
	dfs1(1,0),dfs2(1,0);
	for(i=1;i<=n;i++)printf("%d ",ans[i]?1:0);
	return 0;
}