【国集作业】【AGC010F】Tree Game【博弈论】

洛谷题目链接
小视野链接

题目大意：

给你一棵无根树，树上有一个棋子，每个节点还有若干颗石子，俩人轮流取掉棋子所在节点的一颗石子，并沿树边移动到另一起点，如果轮到一个人执行操作时棋子占据的点上没有石子，那么他就输了。求所有先手必赢的节点。

题解：

博弈论
首先有结论：如果u，v为相邻两个节点，棋子在u节点，并且石子数$a_u>a_v$ 那么先手一定能把后手逼入v节点。
那么，如果一个节点u有子节点v使得$a_u>a_v$且v节点先手必败，那么这节点先手必胜。否则，若先手进入先手必胜的子节点，就把胜利给了对方；若u的子节点v满足$a_u<=a_v$ 则后手可以把先手逼出v节点，使先手无路可走，此时先手必败。
代码就很简单了。。。

#include<iostream>
#include<iomanip>
#include<stack>
#include<queue>
#include<list>
#include<vector>
#include<set> 
#include<map>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<ctime>
#define ll long long
#define db long double
#define inf 5000001
#define INF (ll)1e15 
#define pi acos(-1)
#define mxbit 30
#define rd(n) {n=0;char ch;int f=0;do{ch=getchar();if(ch=='-'){f=1;}}while(ch<'0'||ch>'9');while('0'<=ch&&ch<='9'){n=(n<<1)+(n<<3)+ch-48;ch=getchar();}if(f)n=-n;}
using namespace std;

struct edge{
    int y,nxt;
    edge(){}
    edge(int yy,int nn){
        y=yy,nxt=nn;
    }
}e[inf*2];
int head[inf],ecnt;

void addedge(int x,int y){
    ecnt++;
    e[ecnt]=edge(y,head[x]);
    head[x]=ecnt;
    return;
}

int p[inf],cnt[inf];

void dfs(int u,int last){
    p[u]=0;
    for (int i=head[u];i;i=e[i].nxt){
        int v=e[i].y;
        if (v!=last && cnt[u]>cnt[v]){
            dfs(v,u);
            if (!p[v]){
                p[u]=1;
                break;
            }
        }
    }
    return;
}

int main(){
    int n;
    rd(n)
    for (int i=1;i<=n;i++){
        rd(cnt[i])
    }
    int x,y;
    for (int i=1;i<n;i++){
        rd(x) rd(y)
        addedge(x,y);
        addedge(y,x);
    }
    for (int i=1;i<=n;i++){
        dfs(i,0);
        if (p[i]){
            printf("%d ",i);
        }
    }
    return 0;
}