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;
}
完成啦~下次见