[2017湖南集训7-9]营养餐 阶梯博弈

设num[i]表示点i还能取走多少个水果，num[i]=a[i]-sigma(a[son]*b[son])，
可以发现点i拿走k个后，num[i]-=k，num[fa[i]]+=k*b[i]，于是就能想到阶梯博弈了，先把b[i]=0的点与fa[i]的边断掉，因为取了i点的水果对fa[i]是没有影响的，剩下的森林里，SG函数值即为对奇数层的Nim，即numi的异或和。因为对手把偶数层的移到奇数层，只要把那些石子继续移到下一个奇数层即可。

代码：

#include<iostream>
#include<cstdio>
#include<ctime>
#include<algorithm>
#include<cstring>
#define ll long long
using namespace std;
const int maxn=50010;
int a[maxn],b[maxn],n=0;
ll ans=0;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct edge
{
    int t;
    edge *next; 
}*con[maxn];
void ins(int x,int y)
{
    edge *p;
    p=new edge;
    p->t=y;
    p->next=con[x];
    con[x]=p;
}
void dfs(int v,int fa,int d)
{
    if(b[v]==0) d=1;
    ll dw=0;
    for(edge *p=con[v];p!=NULL;p=p->next)
        if(p->t!=fa) 
        {
            dfs(p->t,v,d+1);
            dw+=a[p->t]*b[p->t];
        }   
    dw=a[v]-dw; 
    if(d&1) ans^=dw;    
}

int main()
{
    freopen("meal.in","r",stdin);
    freopen("meal.out","w",stdout);
    srand(time(NULL));
    int ca=read();
    while(ca--)
    {
        for(int i=1;i<=n;i++)
            con[i]=NULL;
        n=read();
        for(int i=1;i<=n;i++)
            a[i]=read();
        for(int i=1;i<=n;i++)
            b[i]=read();
        for(int i=1;i<=n-1;i++)
        {
            int x=read(),y=read();
            ins(x,y);
            ins(y,x);
        }
        ans=0;
        dfs(1,0,1);     
        if(ans==0) puts("NO");
        else puts("YES");
    }
    return 0;
}