Codeforces Round #417 (Div. 2) E. Sagheer and Apple Tree(树上Nim)

题目链接：Codeforces Round #417 (Div. 2) E. Sagheer and Apple Tree

题意：

给你一棵树，每个节点有a[i]个苹果，有两个人要在这个树上玩游戏。

两个人轮流操作，谁不能操作谁就输了。

这个树有一个特性：叶子到根的距离的奇偶性相同。

每次操作可以选一个节点i，和一个数x,x小于当前节点i的苹果数。

对于节点i，如果是叶子节点，就将这x个苹果吃掉。

如果是非叶子节点，就将这x个苹果移向节点i的任意儿子节点。

现在第二个操作的人要交换两个节点的苹果，使其能赢，

问有多少种交换方式，必须交换。

题解：

仔细分析一下，可以得到是否能赢只与节点深度的奇偶性与叶子节点相同的节点有关。

我们把这些节点放入一个集合S.

所以先将这些节点进行异或一下，得出一个val。如果val等于0，那么当前局势就能赢。

现在来考虑交换节点的苹果。

显然我们要将这个val变成0,所以对S中所有的数都进行枚举一下，然后找不在这个集合中，并且交换后能使val变成0的数。

如果val最开始就为0，那么还要加上S中可以互相交换的个数。和S外可以互相交换的个数。

复杂度O(n)。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;
typedef long long ll;
const int N=1e5+7;

vector<int>g[N];
int cnt[1<<24],cnt2[1<<24],n;
int a[N],x,is,val,have[N],ed;

void dfs(int x=1,int d=0,int fa=0)
{
    if(g[x].size()==0){is=(d&1);}
    for(auto &it:g[x])if(it!=fa)dfs(it,d+1,x);
}

void dfs2(int x=1,int d=0,int fa=0)
{
    if((d&1)==is)have[++ed]=a[x];
    for(auto &it:g[x])if(it!=fa)dfs2(it,d+1,x);
}

int main()
{
    scanf("%d",&n);
    F(i,1,n)scanf("%d",a+i),cnt[a[i]]++;
    F(i,2,n)
    {
        scanf("%d",&x);
        g[x].push_back(i);
    }
    dfs(),dfs2();
    F(i,1,ed)val^=have[i],cnt2[have[i]]++;
    ll ans=0;
    F(i,1,ed)
    {
        int tmp=val^have[i];
        ans+=cnt[tmp]-cnt2[tmp];
    }
    if(!val)
    {
        ans+=1ll*ed*(ed-1)/2;
        ans+=1ll*(n-ed)*(n-ed-1)/2;
    }
    printf("%lld
",ans);
    return 0;
}