题目大意:
给你一棵无根树,树上有一个棋子,每个节点还有若干颗石子,俩人轮流取掉棋子所在节点的一颗石子,并沿树边移动到另一起点,如果轮到一个人执行操作时棋子占据的点上没有石子,那么他就输了。求所有先手必赢的节点。
题解:
博弈论
首先有结论:如果u,v为相邻两个节点,棋子在u节点,并且石子数
au>av
a
u
>
a
v
那么先手一定能把后手逼入v节点。
那么,如果一个节点u有子节点v使得
au>av
a
u
>
a
v
且v节点先手必败,那么这节点先手必胜。否则,若先手进入先手必胜的子节点,就把胜利给了对方;若u的子节点v满足
au<=av
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;
}