AT2307 [AGC010F] Tree Game

https://www.luogu.com.cn/problem/AT2307

考虑枚举一个点$u$做根
那么一定是走向一棵子树$v$,不然没有意义

如果$v$是必败的，后手一定会移回$u$,所以必须要满足$a[u]>a[v]$才能移向$v$

推出这两点就可以做的
code:

#include<bits/stdc++.h>
#define N 3005
using namespace std;
struct edge {
    int v, nxt;
} e[N << 1];
int p[N], eid;
void init() {
    memset(p, -1, sizeof p);
    eid = 0;
}
void insert(int u, int v) {
    e[eid].v = v;
    e[eid].nxt = p[u];
    p[u] = eid ++;
}
int n, a[N];
int dfs(int u, int fa) {
    for(int i = p[u]; i + 1; i = e[i].nxt) {
        int v = e[i].v;
        if(v == fa) continue;
        if(!dfs(v, u) && a[v] < a[u]) return 1;
    }
    return 0;
}
int main() {
    init();
    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);
        insert(u, v), insert(v, u);
    }
    for(int i = 1; i <= n; i ++) if(dfs(i, i)) printf("%d ", i);
    return 0;
}