题意:BC76div1的B题。
分析:一个很明显的树DP,我们设tr[i]表示以i为根的子树方案数,sum[i]表示以i为根对答案做的贡献。但是!在计算贡献的时候不能先把tr[i]乘出来,然后分别求各个儿子的贡献,因为可能会出现tr[i]==0的情况,这样逆元是无效的。我们应该在统计儿子x的时候就算出儿子x的子树方案对当前父亲的贡献的贡献和父亲当前子树方案对儿子x的贡献的贡献。。详见代码。O(n)
代码:
#include<map>
#include<set>
#include<cmath>
#include<queue>
#include<bitset>
#include<math.h>
#include<cstdio>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
const int N=200100;
const int MAX=151;
const int MOD1=1000007;
const int MOD2=100000009;
const double EPS=0.00000001;
typedef long long ll;
const ll MOD=1000000007;
const ll INF=10000000010;
typedef double db;
typedef unsigned long long ull;
int tot,u[N],v[N],pre[N];
void add(int x,int y) {
v[tot]=y;pre[tot]=u[x];u[x]=tot++;
}
ll ans,tr[N],sum[N];
void dfs(int x) {
tr[x]=1;sum[x]=1;
for (int i=u[x];i!=-1;i=pre[i]) {
dfs(v[i]);
sum[x]=((tr[v[i]]+1)*sum[x]+tr[x]*sum[v[i]])%MOD;
tr[x]=tr[x]*(tr[v[i]]+1)%MOD;
}
}
int main()
{
int i,n,x,t;
scanf("%d", &t);
while (t--) {
scanf("%d", &n);
memset(tr,0,sizeof(tr));
memset(sum,0,sizeof(sum));
tot=0;memset(u,-1,sizeof(u));
for (i=2;i<=n;i++) {
scanf("%d", &x);add(x,i);
}
ans=0;dfs(1);
for (i=1;i<=n;i++) ans=(ans+sum[i])%MOD;
printf("%I64d\n", (ans+MOD)%MOD);
}
return 0;
}