Tree 牛客

题目大意：
给定一棵树，对于每个点求包含该点的连通点集的数量

输入
n,n-1条边

输出:
每个点的结果对1e9+7取余

分析:
先只向下考虑，只考虑子树的情况,
设f[x] 以x为根节点子树的联通子集个数
设y_i是x的子节点
f[x] = (f[y_1]+1)(f[y_2]+1)…(f[y_m]+1) (+1是不选这可子树的情况)
然后再考虑计算答案，由向上部分的贡献*子树除了自身的贡献,ans[x]表示x节点的答案,(y是x的一个儿子)
ans[y] = (dp[y]*ans[x]+1) / (dp[y]+1) ,除的部分用逆元算

然后还有一个坑，如果dp[y]+1 是1e9+7的倍数,用之前的办法暴力算

#include<bits/stdc++.h>
using namespace std;
using i64 = long long;
using i128 = __int128;
#define ios ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);

const int mod = 1e9+7;
const int maxn = 1e6+10;
i64 qp(i64 a,i64 b){
    i64 res = 1;
    while(b){
        if(b&1) res = (res*a)%mod;
        b>>=1;
        a = (a*a)%mod;
    }
    return res;
}
int n;
vector<int>g[maxn];
i64 f[maxn];//f[i]表示包含i的连通点集
void dfs1(int x,int fa){
    f[x]=1;
    for(int y :g[x]){
        if(y==fa) continue;
        dfs1(y,x);
        f[x] = (f[x]*(f[y]+1))%mod;
    }
}
i64 ans[maxn];
void dfs2(int x,int fa){
    if(fa==0)ans[x]=f[x];
    else if((f[x]+1)%mod==0){
        dfs1(x,0);
        ans[x]=f[x];
    }
    else ans[x]=(ans[fa]%mod*qp(f[x]+1,mod-2)+1)%mod*(f[x])%mod;
    for(auto y:g[x]){
        if(y==fa)continue;
        dfs2(y,x);
    }
}
int main(){
    ios;
    cin>>n;
    for(int i =1;i<n;++i){
        int x,y;
        cin>>x>>y;
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs1(1,0);
    dfs2(1,0);
    for(int i = 1;i<=n;++i) cout<<ans[i]<<"\n";
    return 0;
}