AGC010 F Tree Game

题意

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

题解

定义np(i)表示只能在i的子树内游戏的必胜必败状态
那么怎么转移呢
其实最关键的东西是石子的限制
首先，肯定只有当np(v)=0时会尝试转移去v，但是后手并不想输，所以它的策略只有搞回去。
所以，如果$a_u>a_v$，那么先手就可以成功塞进v然后获胜，否则会被怼回来，只能从其他地方转移
所以我们枚举一个rt，然后np(rt)就是答案

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
using namespace std;
const int N=1e6+5;
typedef long long ll;
int n,a[N];
struct node{
    int u,v,nxt;
}edge[N*2];
int head[N],mcnt;
void add_edge(int u,int v){
    mcnt++;
    edge[mcnt].u=u;
    edge[mcnt].v=v;
    edge[mcnt].nxt=head[u];
    head[u]=mcnt;
}
bool NP(int u,int fa){
    for(int i=head[u];i;i=edge[i].nxt){
        int v=edge[i].v;
        if(v==fa||a[v]>=a[u])
            continue ;
        if(!NP(v,u))
            return true;
    }
    return false;
}
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);
        add_edge(u,v);
        add_edge(v,u);
    }
    for(int i=1;i<=n;i++){
        if(NP(i,0))
            printf("%d ",i);
    }
}