牛客训练营4 树上博弈

题目传送门
题意，给一颗树，两个人自由选择起点，甲先走，乙后走，输出甲必赢的情况总数。

思路
约定：下文所说距离是默认边权为1。
因为在一颗树上走，那么肯定没有环。
要想甲必赢，那么甲就要一直向乙逼近，让乙走到根节点或者叶节点上，并且这时甲刚好堵住乙，即甲乙相邻。
甲乙相邻时，是甲走完后，才相邻，那么就是赢之前，甲乙距离为2。
再向后推，因为他是一颗树，没有环，所以甲一直向乙逼近，即只要甲乙初始时距离是偶数即可。
我们只要统计节点的深度，深度差为偶数时，距离便是偶数。
对于同一深度产生的贡献是n*n-n，即在此深度选择一个点，到其余n-1个点的距离都是偶数，这一深度对其他深度的影响是奇偶性相同的深度的元素总数减去这个深度的元素总数的积再乘2。

代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const double EPS=1e-6;
typedef long long ll;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
//#define int long long
ll a[N],sum[N],cnt[2]={0};
signed main()
{
    IOS;
    //freopen("","r",stdin);
    //freopen("","w",stdout);
    int n;
    cin>>n;
    a[1]=1;
    cnt[1]=1;
    sum[1]=1;
    ll deep=-1;
    for(int i=2;i<=n;i++)
    {
        int x;
        cin>>x;
        a[i]=a[x]+1;//深度
        deep=max(deep,a[i]);//最大深度
        sum[a[i]]++;//该深度多了一个元素
        cnt[a[i]%2]++;//该奇偶性的深度多了一个元素
    }
    ll res=0;
    for(int i=1;i<=deep;i++)
    {
        if(sum[i])
            res+=sum[i]*sum[i]-sum[i];//同一深度
        cnt[i%2]-=sum[i];
        res+=2*sum[i]*cnt[i%2];//不同深度
    }
    cout<<res<<endl;
}