AtCoder Beginner Contest 160 F - Distributing Integers (product prefix, the root change)

Title: https://atcoder.jp/contests/abc160/tasks/abc160_f

Meaning of the questions: Given a tree, asking each node to start a topological sorting of how many;

Analysis: The whole situation is n! , Which then have to pick out the legal requirements as topological sort, it is required for the first node to the root, so it is from sz [u] to select a program, i.e., multiplied by C (sz [u], 1) for each node recursion also when such a situation. So really want to maintain the size of the plot is the prefix of either.

#include<bits/stdc++.h>
using namespace std;
#define lson root<<1,l,midd
#define rson root<<1|1,midd+1,r
#define pb push_back
typedef long long ll;
const int M=1e6+6;
const int mod=1e9+7;
vector<int>g[M];
ll facn=1;
int n;
ll ans[M],sz[M],mul[M],fac[M];
ll ksm(ll a,ll b){
    a%=mod;
    ll t=1;
    while(b){
        if(b&1)
            t=t*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return t;
}
void dfs1(int u,int fa){
    sz[u]=mul[u]=1;
    for(auto v:g[u]){
        if(v!=fa){
            dfs1(v,u);
            sz[u]+=sz[v];
            mul[u]=mul[u]*mul[v]%mod;
        }
    }
    mul[u]=mul[u]*sz[u]%mod;
}
void dfs2(int u,int fa,ll last){
    ll tmp=last;
    for(auto v:g[u])
        if(v!=fa)
            tmp*=mul[v],tmp%=mod;
    ans[u]=facn*ksm(tmp,mod-2)%mod;
    for(auto v:g[u])
        if(v!=fa)
            last*=mul[v],last%=mod;
    for(auto v:g[u])
        if(v!=fa)
            dfs2(v,u,last*ksm(mul[v],mod-2)%mod*(n-sz[v])%mod);
}
int main(){
    scanf("%d",&n);
    for(ll i=2;i<n;i++)
        facn*=i,facn%=mod;
    for(int u,v,i=1;i<n;i++){
        scanf("%d%d",&u,&v);
        g[u].pb(v);
        g[v].pb(u);
    }
    dfs1(1,0);
    dfs2(1,0,1);
    for(int i=1;i<=n;i++)
        printf("%lld\n",ans[i]);
    return 0;
}
View Code

 

Guess you like

Origin www.cnblogs.com/starve/p/12626868.html