[牛客] Tree-换根dp（预防加法取模导致等0）

链接：https://ac.nowcoder.com/acm/problem/19782
题目：修修去年种下了一棵树，现在它已经有n个结点了。
        修修非常擅长数数，他很快就数出了包含每个点的连通点集的数量。
        澜澜也想知道答案，但他不会数数，于是他把问题交给了你。

思路是很简单的换根dp，先求出以1为根的值，然后进行撤回操作再换根，这里最要注意的坑点是 dp【v】+1%mod会等于0（加法会导致模等于0），所以我们这里需要特判处理。

#include <iostream>
#include <cstdio>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <deque>
#include <vector>
#include <queue>
#include <string>
#include <cstring>
#include <map>
#include <stack>
#include <set>
#include <cstdlib>
#define INF 0x3f3f3f3f3f3f3f3f
#define inf 0x3f3f3f3f
#define FILL(a,b) (memset(a,b,sizeof(a)))
#define re register
#define lson rt<<1
#define rson rt<<1|1
#define lowbit(a) ((a)&-(a))
#define ios std::ios::sync_with_stdio(false);std::cin.tie(0);std::cout.tie(0);
#define fi first
#define rep(i,n) for(int i=0;(i)<(n);i++)
#define rep1(i,n) for(int i=1;(i)<=(n);i++)
#define se second
#define scd(a) scanf("%d",&a)
#define scdd(a,b) scanf("%d%d",&a,&b)
#define scddd(a,b,c) scanf("%d%d%d",&a,&b,&c)
#define ac cout<<ans<<"\n"
#define F(x) ((x)/3+((x)%3==1?0:tb))
#define G(x) ((x)<tb?(x)*3+1:((x)-tb)*3+2)
using namespace std;
typedef long long  ll;
typedef unsigned long long  ull;
typedef pair<ll,ll> pii;
int dx[4]= {-1,1,0,0},dy[4]= {0,0,1,-1};
const ll mod=1e9+7;
const ll N =1e6+10;
const double eps = 1e-4;
//const double pi=acos(-1);
ll qk(ll a,ll b){
    ll ans=1;
    while(b){
        if(b&1) ans=(ans*a)%mod;
        a=(a*a)%mod;
        b/=2;
    }
    return ans%mod;
}
int n;
ll dp[N];
vector<int> g[N];
ll w[N],f[N];
void dfs(int u,int f){
    dp[u]=1;
    for(int v:g[u]){
        if(v==f) continue;
        dfs(v,u);
        dp[u]=(dp[u]*((dp[v]+1ll)%mod))%mod;
    }
}
void DP(int u,int p){
    for(int v:g[u]){
        if(v==p) continue;
        if((dp[v]+1)%mod==0){
           if(p) f[v]=f[u]+1;
           else f[v]=1;
           for(int k:g[u]){
               if(k==v||k==p) continue;
               f[v]=(f[v]*(dp[k]+1)%mod)%mod;
           }
        }
        else f[v]=(w[u]*qk(dp[v]+1ll,mod-2))%mod;
        w[v]=(dp[v]*((f[v]+1ll)%mod))%mod;
        DP(v,u);
    }
}
void sovle(){
    cin>>n;
    for(int i=2;i<=n;i++){
        int u,v;
        cin>>u>>v;
        g[u].push_back(v);
        g[v].push_back(u);
    }
    dfs(1,0);
    w[1]=dp[1];
    DP(1,0);
    for(int i=1;i<=n;i++) cout<<w[i]<<endl;
}
int main()
{
#ifdef LOCAL
    freopen("in.txt", "r", stdin);
#else
    int t=1;
   //cin>>t;
    while(t--) sovle();
#endif // LOCAL
    return 0;
}