제출 #1340245

#제출 시각아이디문제언어결과실행 시간메모리
1340245trungcanDrzava (COCI25_drzava)C++20
110 / 110
460 ms36076 KiB
#include <bits/stdc++.h>

using namespace std;

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

long long ans[N];
int dp[N][N];
short sz[N], 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 + j] + dp[u][i] * 1LL * dp[v][j]) % MOD;
            sz[u] += sz[v];
        }
    dp[u][1] = sz[u];
}

void reroot(short u, short par){
    ans[2] += dp[u][1] - 1;
    for (short i = 2; i < n; ++i)
        ans[i + 1] += dp[u][i];
    vector<int> f[2];
    short sc = sz[u];
    for (short i = 0; i <= sz[u]; ++i) f[0].push_back(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){
            int tmp = sz[v];
            f[1].clear();
            for (short i = 0; i <= sz[v]; ++i) f[1].push_back(dp[v][i]);

//            cout << v << "\t"; for (int i = 0; i <= sz[u]; ++i) cout << dp[u][i] << " "; cout << "\n";
            dp[u][1] = sz[u] - sz[v] - 1;
            for (short i = 2; i <= sz[u] - sz[v]; ++i)
                for (short j = 1; j <= min(i, sz[v]); ++j)
                    dp[u][i] = (dp[u][i] - (dp[u][i - j] * 1LL * dp[v][j]) % MOD + MOD) % MOD;
            for (short i = sz[u] - sz[v] + 1; i <= sz[u]; ++i) dp[u][i] = 0;
            sz[u] = sc - sz[v]; ++dp[u][1];

//            cout << v << "\t"; for (int i = 0; i <= sz[u]; ++i) cout << dp[u][i] << " "; cout << "\n";

            --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 + j] + dp[u][j] * 1LL * dp[v][i]) % MOD;
            sz[v] += sz[u]; dp[v][1] = sz[v];

            reroot(v, u);

            sz[v] = tmp; sz[u] = sc;
            for (short i = 0; i <= sz[u]; ++i) dp[u][i] = f[0][i];
            for (short i = 0; i <= sz[v]; ++i) dp[v][i] = f[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].emplace_back(v);
        adj[v].emplace_back(u);
    }

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

    ans[1] = n;
    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...