2020牛客寒假算法基础集训营4 F题

2020牛客寒假算法基础集训营4

F题 树上博弈

又开始补题啦。这一题算是思维把=-=
传送门

题目的大概意思就是有一棵树，然后两个人在上面玩游戏，牛牛先开始游戏，牛妹再开始游戏。每次只能移动一步，谁先无路可走谁就输，问现在可以选择牛牛和牛妹的起始位置，有多少种不同的开始方式可以让牛牛最终获胜，输出数量

题目的大概思路：我们画个图其实就可以发现，一方想要赢，必须把对方逼上绝路，就是叶子结点，把对方堵死了，无路可走了，就获胜了。我们还可以发现：自己与对方的相差的距离如果是奇数，那么自己就会输掉，如果自己与对方相差的距离为偶数，那么自己就会获胜。(这种情况下最终其实是可以把对方逼上叶子结点的）
那我们只需要统计出偶数深度的结点和奇数深度的结点有多少个（这里的深度是相对于根结点而言的），然后两种情况，第一种：都位于奇数深度的结点，相差的距离为偶数，那么牛牛获胜，假设奇数深度的结点有n个，那个牛牛选择一个位置后，还有n-1个位置牛妹选择使得牛牛获胜，共有n*（n-1）种情况；第二种:都位于偶数深度的结点，相差的距离也为偶数，那么牛牛获胜。计算方法同上。

上代码啦~

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6 + 10;

int n;
int d[N];//记录深度
int cnt[N];//记录奇数深度和偶数深度的结点个数
ll ans;

int main()
{
	cin >> n;
	d[1] = 0;
	cnt[0]++;
	
	for (int i = 2; i <= n; i++)
	{
		int temp;
		cin >> temp;
		d[i] = (d[temp] + 1) % 2;
		cnt[d[i]]++;
	}
	ans += 1ll * cnt[0] * (cnt[0] - 1) + 1ll * cnt[1] * (cnt[1] - 1);
	cout << ans << endl;
	return 0;
}

完成啦~下次见