[树形dp + 换根]Tree

题目：Tree

分析：

这个题第一步很好想，求出每个子树的答案。难的是如何求u上方的贡献。

（引用）

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;//三年竞赛一场空，不开long long见祖宗 
//typedef __int128 lll;
#define print(i) cout << "debug: " << i << endl
#define close() ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
#define mem(a, b) memset(a, b, sizeof(a))
#define pb(a) push_back(a)
#define x first
#define y second
typedef pair<int, int> par;
const ll mod = 1e9 + 7;
const int maxn = 1e6 + 10;
const int inf = 0x3f3f3f3f;
int n;
vector<int> g[maxn];
ll dp[maxn];
ll up[maxn], res[maxn];
int f[maxn];

ll qpow(ll a, ll n)
{
    ll res = 1;
    while(n)
    {
        if(n & 1) res = res * a % mod;
        a = a * a % mod, n >>= 1;
    }
    return res;
}

void dfs1(int u, int fa)
{
    dp[u] = 1;
    f[u] = fa;
    for(int v : g[u])
    {
        if(v == fa) continue;
        dfs1(v, u);
        dp[u] = dp[u] * (dp[v] + 1) % mod;
    }
}

void dfs2(int u, int fa)
{
    if(u != 1)
    {
        if((dp[u] + 1) % mod == 0)
        {
            up[u] = up[fa] + 1;
            for(int v : g[fa])
                if(v != u && v != f[fa])
                    up[u] = up[u] * (dp[v] + 1) % mod;
        }
        else up[u] = res[fa] * qpow(dp[u] + 1, mod - 2) % mod;
        res[u] = (up[u] + 1) * dp[u] % mod;
    }
    else res[u] = dp[u];
    for(int v : g[u])
        if(v != fa)
            dfs2(v, u);
}

int main()
{
    cin >> n;
    for(int i = 1; i < n; i++)
    {
        int a, b; cin >> a >> b;
        g[a].pb(b), g[b].pb(a);
    }
    dfs1(1, -1);
    dfs2(1, -1);
    for(int i = 1; i <= n; i++)
        printf("%lld\n", res[i]);
}