【AtCoder2307】Tree Game（博弈）

题意

有一棵树，每个结点上有A[i]个石子，A开始把一个棋子放在任意一个结点上，然后由A开始，A,B轮流操作
将当前棋子的结点石子数减1
将棋子移动到相邻的结点上
不能操作者输
求A开始把棋子放在哪儿保证能赢。

题解

性质：棋子不会往比自己石子多的地方走（如果这样走，对方可以再走回来，自己就少了一颗棋子，对方又比自己多，打消耗战打不赢的。。。）
然后就可以直接模拟计算NP状态
枚举一个点作为起点，然后dfs遍历，求出所有NP

代码

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int MAXN=3005;

int n,A[MAXN];
vector<int> adj[MAXN];
bool st[MAXN];

void dfs(int u,int fa=0)
{
	st[u]=false;
	A[u]--;
	for(int i=0;i<(int)adj[u].size();i++)
	{
		int v=adj[u][i];
		if(v==fa||A[v]>A[u])
			continue;
		dfs(v,u);
		if(st[v]==false)
		{
			st[u]=true;
			break;
		}
	}
	A[u]++;
}

int main()
{
	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);
		adj[u].push_back(v);
		adj[v].push_back(u);
	}
	for(int i=1;i<=n;i++)
	{
		dfs(i);
		if(st[i])
			printf("%d ",i);
	}
	puts("");
	
	return 0;
}