Submission #1340227

#TimeUsernameProblemLanguageResultExecution timeMemory
1340227trungcanDrzava (COCI25_drzava)C++20
18 / 110
3 ms2372 KiB
#include <bits/stdc++.h>
#define ll long long

using namespace std;

const int N = 5005;
const int MOD = 1e9 + 7;
const ll LIM = 2e18;

ll dp[N][N], f[N][2][N], ans[N];
short sz[N], s[N][2];
short id[N], cnt, n;
vector<short> adj[N];

void dfs(short u, short par){
    dp[u][0] = sz[u] = 1;
    for (short v: adj[u])
        if (v != par){
            dfs(v, u);

            for (short i = sz[u]; i >= 0; --i)
                for (short j = 1; j <= sz[v]; ++j){
                    dp[u][i + j] += (dp[u][i] * 1LL * dp[v][j]) % MOD;
                    if (dp[u][i + j] >= LIM) dp[u][i + j] %= MOD;
                }
            sz[u] += sz[v];
        }
    dp[u][1] = sz[u];
}

void reroot(short u, short par){
    for (short i = 0; i < n; ++i)
        ans[i + 1] += dp[u][i] % MOD + (i == 1 ? -1 : 0);
    id[u] = ++cnt;
//    cout << "\t\t" << u << "\n";
//    for (int i = 1; i <= n; ++i) cout << i << " " << sz[i] << "\n";
//    for (int i = 0; i <= sz[u]; ++i) cout << dp[u][i] << " "; cout << '\n';
    for (short v: adj[u])
        if (v != par){
            s[id[u]][0] = sz[u]; s[id[u]][1] = sz[v];

            for (short i = 0; i <= sz[u]; ++i) f[id[u]][0][i] = dp[u][i], dp[u][i] = 0;
            for (short i = 0; i <= sz[v]; ++i) f[id[u]][1][i] = dp[v][i];


            sz[u] = dp[u][0] = 1;
            for (short w: adj[u])
                if (w != v){
                    for (short i = sz[u]; i >= 0; --i)
                        for (short j = 1; j <= sz[w]; ++j){
                            dp[u][i + j] += (dp[u][i] * 1LL * dp[w][j]) % MOD;
                            if (dp[u][i + j] >= LIM) dp[u][i + j] %= MOD;
                        }
                    sz[u] += sz[w];
                }
            dp[u][1] = sz[u];

            --dp[v][1];
            for (short i = sz[v]; i >= 0; --i)
                for (short j = 1; j <= sz[u]; ++j){
                    dp[v][i + j] += (dp[v][i] * 1LL * dp[u][j]) % MOD;
                    if (dp[v][i + j] >= LIM) dp[v][i + j] %= MOD;
                }
            sz[v] += sz[u]; dp[v][1] = sz[v];

            reroot(v, u);

            sz[u] = s[id[u]][0]; sz[v] = s[id[u]][1];
            for (short i = 0; i <= sz[u]; ++i) dp[u][i] = f[id[u]][0][i];
            for (short i = 0; i <= sz[v]; ++i) dp[v][i] = f[id[u]][1][i];
        }
}

int main(){
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);

    cin >> n;
    for (short i = 1, u, v; i < n; ++i){
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }

    dfs(1, 1);
    reroot(1, 1);

    for (short i = 1; i <= n; ++i) cout << ans[i] % MOD << " ";
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...