BC#76 C_HDU5647 DZY Loves Connecting 树形DP

来源：互联网发布：系统还原数据恢复编辑：程序博客网时间：2024/05/22 02:27

比赛的时候没想出来，现在发现还是一个经典的树形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 fusing 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;}

0 0