hdu5647DZY Loves Connecting

来源：互联网发布：台湾中央研究院知乎编辑：程序博客网时间：2024/06/16 05:41

链接：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;}

0 0