Submission #1340223

#TimeUsernameProblemLanguageResultExecution timeMemory
1340223trungcanDrzava (COCI25_drzava)C++20
61 / 110
2095 ms78260 KiB
#include <bits/stdc++.h>

using namespace std;

const int N = 5005;
const int MOD = 1e9 + 7;

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

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

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

void reroot(int u, int par){
    for (int i = 0; i < n; ++i)
        ans[i + 1] = (ans[i + 1] + dp[u][i] + (i == 1 ? -1 : 0)) % MOD;
    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 (int v: adj[u])
        if (v != par){
            s[id[u]][0] = sz[u]; s[id[u]][1] = sz[v];

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


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

            --dp[v][1];
            for (int i = sz[v]; i >= 0; --i)
                for (int j = 1; j <= sz[u]; ++j)
                    dp[v][i + j] = (dp[v][i + j] + (dp[v][i] * 1LL * dp[u][j]) % MOD) % 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 (int i = 0; i <= sz[u]; ++i) dp[u][i] = f[id[u]][0][i];
            for (int 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 (int 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 (int i = 1; i <= n; ++i) cout << ans[i] << " ";
}
#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...