bzoj 5356: 红与蓝

题意：

给定一棵树，初始时非叶节点均为无色，叶节点会是红色、蓝色或无色。小红和小蓝轮流给无色叶子染色（小红染红色，小蓝染蓝色，小红先染）。所有叶子染完后，非叶节点的颜色将被逐一确定：一个非叶节点的颜色是它所有儿子的颜色中出现较多的那个（保证有奇数个儿子）。最后，根是谁的颜色谁就获胜。求小红是否能赢，若能赢，求出第一步选择哪些叶子能赢。

题解：

每个点有3种情况：红胜，蓝胜（某种确定颜色比另一种多），先手胜。从下往上树形dp求出每个点的情况。
考虑第二问，如果根是红胜的话，可以选任意一个叶子；如果是先手胜的话，需要将某个蓝胜的子树变成先手胜（这样总体还是先手胜），或是将某个先手胜的子树变成蓝胜。
如果需要将蓝胜变为先手胜，同样考虑改变某个儿子，也有可能无解。
code：

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,c[100010],s[100010],top=0,g[100010],fa[100010];
struct node{
	int y,next;
}a[200010];int len=0,last[100010];
void ins(int x,int y)
{
	a[++len].y=y;
	a[len].next=last[x];last[x]=len;
}
void dfs(int x)
{
	if(c[x]!=-1) return;
	int s1=0,s2=0;
	for(int i=last[x];i;i=a[i].next)
	{
		int y=a[i].y;
		dfs(y);fa[y]=x;
		if(c[y]==-1) continue;
		if(c[y]==0) s1++;
		else s2++;
	}
	if(s1>s2) c[x]=0;
	if(s1<s2)
	{
		c[x]=1;
		if(s1+1==s2) c[x]=2;
	}
}
void in(int x)
{
	if(c[x]!=-1&&c[x]!=2) return;
	if(!last[x]) s[++top]=x;
	for(int i=last[x];i;i=a[i].next) in(a[i].y);
}
int main()
{
	int T;scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		len=0;memset(last,0,sizeof(last));
		for(int i=1;i<=n;i++)
		{
			int f;scanf("%d",&f);
			if(i!=1) ins(f,i);
		}
		for(int i=1;i<=n;i++) scanf("%d",&c[i]),g[i]=c[i];
		dfs(1);
		int s1=0,s2=0;
		for(int i=last[1];i;i=a[i].next)
		{
			int y=a[i].y;
			if(c[y]==-1) continue;
			if(c[y]==0) s1++;
			else s2++;
		}
		if(s1==s2)
		{
			top=0;in(1);
			sort(s+1,s+top+1);
			printf("%d ",top);
			for(int i=1;i<=top;i++) printf("%d ",s[i]);printf("\n");
		}
		else if(s1>s2)
		{
			top=0;
			for(int i=1;i<=n;i++)
				if(!last[i]&&c[i]==-1) s[++top]=i;
			printf("%d ",top);
			for(int i=1;i<=top;i++) printf("%d ",s[i]);printf("\n");
		}
		else printf("-1\n");
	}
}