BC#76 C_HDU5647 DZY Loves Connecting 树形DP

比赛的时候没想出来，现在发现还是一个经典的树形DP

首先用size[i]表示节点i对答案的贡献

同时为了推倒方便，用f[i]表示以i为根的子树的数目

那么我们考虑当i的子节点j加入时，对答案的贡献

首先，加入j时，会有f[i] * f[j]个新的子树加入，那么计算对答案的贡献

之前的size[i]已经被计算过一次，加入j之后，对于答案的贡献是size[i] * f[j]

此外，还需要统计右边的数目那就是f[i] * size[j]

size[i] = size[i] * (1 + f[j]) + size[j] * f[i]就是总的递推式了

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#define ll long long
#define rep(i, j, k) for(int i = j; i <= k; i++)
#define mod 1000000007
#define dp f

using namespace std;

int n, father[200500];
ll size[200500], f[200500], ans = 0;

int main()
{
	int Case;
	cin >> Case;
	while (Case--)
	{
		ans = 0;
		scanf ("%d", &n);
		rep (i, 2, n)
				scanf ("%d", &father[i]);
		rep (i, 1, n)
			f[i] = size[i] = 1;
		for (int i = n; i >= 1; i--)
		{
			int now = father[i];
			size[now] += (size[i] * f[now] % mod) + (size[now] * f[i] % mod);
			size[now] %= mod;
			f[now] += f[now] * f[i] % mod;
			f[now] %= mod;
		}
		rep (i, 1, n)
			ans = (ans + size[i]) % mod;
		cout << ans << endl;
	}
	return 0;
}