AGC010F Tree Game

传送门

题意：
有一棵有n个节点的树。现在第i个节点放有a[i]个石子。Takahashi和Aoki想要用这棵
树玩一个游戏。首先，Takahashi选择一个节点并放一个木块在上面。然后从Takahashi开
始，他们轮流进行如下操作：
从木块所在的节点移除一个石子，然后把木块向相邻的一个节点移动。
若当某个人进行操作的时候当前木块所在节点没有石子，那么他就不能进行这轮操作并
输掉游戏。请找出所有能让Takahashi赢得游戏的放置木块的初始节点，在两人都采用最
优操作的情况下。

题解：
我们可以直接枚举初始节点，并令其为根。
把游戏范围限制在一棵子树里，令f[u]为以u为根的子树下先手是否赢。那么只要有一个儿子v使a[v]<a[u]且f[v]=0，f[u]就为1.否则f[u]为0.
证明：把木块从u移到v，因为a[v]<a[u]，所以对手无论如何都会在v的子树中继续游戏。因为f[v]=0，所以获胜。

代码：

#include<cstdio>
#include<cstring>
#define maxn 3005
int n,a[maxn],ans[maxn],cnt;
bool f[maxn];
struct node { int v; node *nxt; } edge[maxn*2],*head[maxn],*ncnt;
void addedge(int u,int v)
{
	ncnt++;
	ncnt->v=v,ncnt->nxt=head[u];
	head[u]=ncnt;
}
bool dfs(int u,int fa)
{
	for(node *p=head[u];p;p=p->nxt)
	{
		int v=p->v;
		if(a[v]<a[u]&&!dfs(v,u)) return f[u]=true;
	}
	return f[u]=false;
}
int main()
{
	ncnt=&edge[0];
	scanf("%d",&n);
	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		addedge(u,v); addedge(v,u);
	}
	for(int i=1;i<=n;i++)
	{
		memset(f,0,sizeof(f));
		if(dfs(i,0)) ans[++cnt]=i;
	}
	for(int i=1;i<=cnt;i++) printf("%d ",ans[i]);
}