营养餐

题目大意

JM 是 DY 的好朋友。为了感谢 JM 多年来对自己的关心,DY 决定请他吃一顿水果营养餐.
DY 有一棵有 n 个结点的树,结点 1 为根。树上每一个结点都长着许多水果,其中,结点 i 上有 ai 个水果,每个水果重 bi .
水果虽然好吃,但是这棵树非常脆弱! 一旦某结点的子结点上的水果总重量过大,树枝就会承受不住压力而断裂! 所以,随时需要保持任意一个结点 i:
ai>=c is childacbc
(数据保证初始局面满足该条件).
JM 说, 谢谢你的这些水果, 不过我的内心毫无波动, 甚至还有点想博弈. 我们轮流来操作: 选择一个结点, 摘走若干个水果, 不可以不摘. 不可操作者输.
DY 说, 这毫无意义,我们都足够聪明,这个游戏一旦确定谁为先手,结果也确定了.
然而, 在前排吃瓜的你并不知道游戏的结果,所以,你得编写程序来知道谁会赢.

做法

不妨记 numi=aij is childajbj
我们注意到如果bi=0就不影响父亲,可以断开其与父亲的连边。
然后可以发现任意一个联通块其实都是阶梯nim游戏。

#include
#include
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=50000+10;
int cnt[1000000+10];
int a[maxn],b[maxn],c[maxn],dep[maxn],size[maxn],siz[maxn];
int h[maxn],go[maxn*2],nxt[maxn*2],du[maxn],dl[maxn],fa[maxn];
int i,j,k,l,t,n,m,wdc,tot,top,ans,ca,head,tail,now;
bool czy;
void add(int x,int y){
    go[++tot]=y;
    nxt[tot]=h[x];
    h[x]=tot;
}
void dfs(int x,int y){
    dep[x]=dep[y]^1;
    int t=h[x],l=a[x];
    size[x]=siz[x]=0;
    while (t){
        if (go[t]!=y){
            dfs(go[t],x);
            if (b[go[t]]!=0){
                size[x]^=size[go[t]];
                siz[x]^=siz[go[t]];
            }
            l-=a[go[t]]*b[go[t]];
        }
        t=nxt[t];
    }
    size[x]^=l;
    if (dep[x]) siz[x]^=l;
    if (!y||b[x]==0) ans^=(!dep[x]?size[x]^siz[x]:siz[x]);
}
int main(){
    freopen("meal.in","r",stdin);freopen("meal.out","w",stdout);
    scanf("%d",&ca);
    while (ca--){
        scanf("%d",&n);
        fo(i,1,n) scanf("%d",&a[i]);
        fo(i,1,n) scanf("%d",&b[i]);
        fo(i,1,n) h[i]=0;
        tot=0;
        fo(i,1,n-1){
            scanf("%d%d",&j,&k);
            add(j,k);add(k,j);
        }
        ans=0;
        dep[0]=1;
        dfs(1,0);
        if (ans) printf("YES\n");else printf("NO\n");
    }
}

你可能感兴趣的:(博弈论)