hdu5647DZY Loves Connecting

链接:http://acm.hdu.edu.cn/showproblem.php?pid=5647

题意: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;
}


你可能感兴趣的:(hdu5647DZY Loves Connecting)