P2796 Facer的程序

最新推荐文章于 2020-07-31 08:41:48 发布

anliang01409077

最新推荐文章于 2020-07-31 08:41:48 发布

阅读量151

点赞数

原文链接：http://www.cnblogs.com/QYJ060604/p/11423693.html

版权

P2796 Facer的程序

树形dp

其实就是求大树中小树的个数

f[i]表示以i为根的答案

f[i]=∏_{j∈i的孩子}(f[j]+1)

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=100005;
typedef long long ll;
ll mod=1000000007ll;
ll f[N]={0};
ll ans=0;
int hed[N<<1],tal[N<<1],nxt[N<<1],cnt=0;
int n;
inline void addege(int x,int y){
    cnt++;
    tal[cnt]=y;
    nxt[cnt]=hed[x];
    hed[x]=cnt;
}
void dfs(int u,int fa){
    f[u]=1ll;
    for(int i=hed[u];i;i=nxt[i]){
        int v=tal[i];
        if(v==fa) continue;
        dfs(v,u);
        f[u]=f[u]*(f[v]+1ll);
        f[u]%=mod;
    }
    ans+=f[u]%mod;
    ans%=mod;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        addege(x,y);
        addege(y,x); 
    }
    dfs(1,1);
    printf("%lld\n",ans);
    return 0;
}