Submission #1161157

#TimeUsernameProblemLanguageResultExecution timeMemory
1161157justRoad Closures (APIO21_roads)C++20
5 / 100
2097 ms29512 KiB
#include "bits/stdc++.h"
using namespace std;

#define all(x) (x).begin(), (x).end()
#define int long long
#define vec vector
#define pii pair<int, int>

#define X first
#define Y second

int dfs(int u, int p, int K, int k, vec<vec<pii>> &adj) {
    vec<pii> vals; vals.reserve(adj[u].size());

    for(auto [v, w]: adj[u]) {
        if (v == p) continue;
        vals.push_back({
            dfs(v, u, K, K - 1, adj) + w,
            dfs(v, u, K, K, adj)
        });
    }

    sort(all(vals), [](pii a, pii b) {
        return a.X - a.Y > b.X - b.Y;
    });

    int res = 0;
    for (auto [a, b]: vals) {
        if (k > 0 && a > b) {
            res += a;
            k--;
        } else {
            res += b;
        }
    }

    return res;
}

vec<int> calc(int n, vec<pii> edges, vec<int> costs) {

    if (n == 2) {
        return {costs[0], 0};
    }


    bool is_star = true;
    for(auto [u, _]: edges) is_star &= u == 0;

    if (is_star) {
        sort(all(costs));
        partial_sum(all(costs), costs.begin());
        reverse(all(costs));
        costs.push_back(0);
        return costs;
    }

    assert(!is_star);

    // bool is_line = true;
    // for (int i = 0; i < n - 1; i++) {
    //     is_line &= edges[i].X == i && edges[i].Y == i + 1;
    // }

    // if (is_line) {
    //     vec<int> ans(n, 0);

    //     int sum = 0;
    //     for(auto x: costs) sum += x;
    //     ans[0] = sum;

    //     vec<vec<int>> dp(n, vec<int>(2, 0));

    //     dp[0][0] = 0;
    //     dp[0][1] = 0;

    //     dp[1][0] = costs[0];
    //     dp[1][1] = 0;

    //     dp[2][0] = min(dp[1][0], dp[1][1]) + costs[1];
    //     dp[2][1] = dp[1][0];

    //     for (int i = 3; i < n; i++) {
    //         dp[i][0] = min(dp[i - 1][0], dp[i - 1][1]) + costs[i - 1];
    //         dp[i][1] = dp[i - 1][0];
    //     }

    //     ans[1] = min(dp[n - 1][0], dp[n - 1][1]);
    //     return ans;
    // }

    // assert(!is_line);

    int sum = 0;
    for(auto x: costs) sum += x;

    
    vec<vec<pii>> adj(n);
    for (int i = 0; i < n - 1; i++) {
        adj[edges[i].X].push_back({edges[i].Y, costs[i]});
        adj[edges[i].Y].push_back({edges[i].X, costs[i]});
    }

    for(auto &x: adj) sort(all(x));

    // kind of inversing the inversing the question
    vec<int> ans(n, sum);
    for(int k = 1; k < n; k++) {
        ans[k] -= dfs(0, -1, k, k, adj);
    }

    return ans;
}

vec<int> minimum_closure_costs(
    signed N, 
    vec<signed> U, 
    vec<signed> V, 
    vec<signed> W
) {
    vec<pii> edges(N - 1);
    vec<int> costs(N - 1);

    for (int i = 0; i < N - 1; i++) {
        edges[i] = {min(U[i], V[i]), max(U[i], V[i])};
        costs[i] = W[i];
    }

    return calc(N, edges, costs);
}


#ifdef debug
signed main() {
    freopen("input.txt", "r", stdin);

    int n; cin >> n;
    vec<signed> u(n - 1), v(n - 1), w(n -1 );
    for (int i = 0; i < n - 1; i++) {
        cin >> u[i] >> v[i] >> w[i];
    }

    vec<int> ans = minimum_closure_costs(n, u, v, w);
    for (int i = 0; i < n; i++) {
        cout << ans[i] << " ";
    }
    cout << endl;

}
#endif
#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...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...