bzoj4134 (sg函数，线段树合并（trie））

 

trie合并，加sg函数。。。

 

这道题体现了dp思想处理博弈论问题，同时发现许多的书上信息合并问题可以考虑线段树合并。

也体现了，trie处理xor问题，也体现了trie处理mex问题。

 

 

 

#include<cstdio>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<stack>
using namespace std;
const int N=100005;
const int D=19;

int n;
int c[N];

int head[N],tot,pre[N*2],to[N*2];
void addedge(int u,int v){to[++tot]=v;pre[tot]=head[u];head[u]=tot;}
int dp[N];
int num,rt[N],ch[N*41][2],rev[N*41],val[N*41];
void build(int &u,int tmp,int D) 
{
	if (u==0) u=++num;
	if (D==-1) {val[u]=1;return ;}
	build(ch[u][tmp>>D&1],tmp,D-1);
}
void down(int u,int D) 
{
	rev[ch[u][0]]^=rev[u];rev[ch[u][1]]^=rev[u];
	if (rev[u]&(1<<D)) swap(ch[u][0],ch[u][1]);
	rev[u]=0;
}
int merge(int u,int v,int D) 
{
	if (!u||!v) return v+u;
	if (D==-1) return val[u]=1,u;
	down(u,D);down(v,D);
	int tmp=++num;
	ch[tmp][0]=merge(ch[u][0],ch[v][0],D-1);
	ch[tmp][1]=merge(ch[u][1],ch[v][1],D-1);
	val[tmp]=val[ch[tmp][0]]&&val[ch[tmp][1]];
	return tmp;
}
int get_sg(int u)
{
	int ans=0;
	for (int i=D;i>=0;i--) 
	{
		down(u,i);
		if (val[ch[u][0]]==0) u=ch[u][0];else u=ch[u][1],ans+=(1<<i); 
	} 
	return ans;
}
int h[N];
void dfs(int u,int fa)
{
	for (int i=head[u];i;i=pre[i]) if (to[i]!=fa)dfs(to[i],u),h[u]^=dp[to[i]];
	if (!c[u]) build(rt[u],h[u],D);
	for (int i=head[u];i;i=pre[i]) if (to[i]!=fa) 
	{
		rev[rt[to[i]]]^=h[u]^dp[to[i]];
		rt[u]=merge(rt[u],rt[to[i]],D);
	}
	dp[u]=get_sg(rt[u]);
}

int ans[N],cnt;
void find_ans(int u,int fa,int now)
{
	now^=dp[u]^h[u];
	if (!now&&!c[u]) ans[++cnt]=u;
	for (int i=head[u];i;i=pre[i])
	if (fa!=to[i])  find_ans(to[i],u,now);
}

int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++) scanf("%d",&c[i]);
	int u,v;
	for (int i=1;i<n;i++) 
	{
		scanf("%d%d",&u,&v);
		addedge(u,v);
		addedge(v,u);
	}
	dfs(1,0);
	if (dp[1]==0) {printf("-1\n");return 0;}
	
	find_ans(1,0,dp[1]);
	
	sort(ans+1,ans+cnt+1);
	for (int i=1;i<=cnt;i++) printf("%d\n",ans[i]);
	return 0;
}