Submission #1340232

#TimeUsernameProblemLanguageResultExecution timeMemory
1340232trungcanDrzava (COCI25_drzava)C++20
61 / 110
2096 ms172340 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)
                    if (dp[u][i] && dp[v][j]){
                        dp[u][i + j] += (dp[u][i] % MOD * 1LL * (dp[v][j] % MOD)) % 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;
    int sc = sz[u];
    for (short i = 0; i <= sz[u]; ++i) f[id[u]][0][i] = dp[u][i];
//    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]][1] = sz[v];

            for (short i = 0; i <= sz[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)
                            if (dp[u][i] && dp[w][j]){
                                dp[u][i + j] += (dp[u][i] % MOD * 1LL * (dp[w][j] % MOD)) % 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)
                    if (dp[v][i] && dp[u][j]){
                        dp[v][i + j] += (dp[v][i] % MOD * 1LL * (dp[u][j] % MOD)) % 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[v] = s[id[u]][1];
            for (short i = 0; i <= sz[v]; ++i) dp[v][i] = f[id[u]][1][i];
        }
  
  sz[u] = sc;
  for (short i = 0; i <= sz[u]; ++i) dp[u][i] = f[id[u]][0][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...