Submission #697608

#	Time	Username	Problem	Language	Result	Execution time	Memory
697608		Dan4Life	Road Closures (APIO21_roads)	C++17	12 / 100	54 ms	6680 KiB

roads

#include "roads.h"
#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define pb push_back
#define sz(a) (int)a.size()
#define all(a) a.begin(),a.end()
using ll = long long;
const int mxN = (int)2e3+10;
const ll LINF = (ll)1e18;

int n, k;
ll dp[mxN][2];
vector<pair<int,ll>> adj[mxN];

void dfs(int s, int p){
    vector<pair<int,ll>> v; v.clear();
    for(auto x : adj[s]){
        int u = x.fi; ll w = x.se;
        if(u!=p) dfs(u,s), v.pb({u,w});
    }
    sort(all(v), [&](pair<int,ll> a, pair<int,ll>b){
        ll x = dp[a.fi][1]-dp[a.fi][0]+a.se;
        ll y = dp[b.fi][1]-dp[b.fi][0]+b.se;
        return x<y;
    });
    int x = 0;
    for(auto y : v){
        int u = y.fi; ll w = y.se;
        if(x<k) dp[s][0]+=min(dp[u][0]+w,dp[u][1]);
        else dp[s][0]+=dp[u][1];
        if(x<k-1) dp[s][1]+=min(dp[u][0]+w,dp[u][1]);
        else dp[s][1]+=dp[u][1];
        x++;
    }
}

vector<ll> minimum_closure_costs(int N, vector<int> U, vector<int> V, vector<int> W) {
    n = N;
    vector<ll> ans(N,0ll);
    ans[0] = accumulate(all(W),0ll);
    if(accumulate(all(U),0ll)==0ll){
        sort(all(W)); reverse(all(W)); int i = 0;
        ll tot = accumulate(all(W),0ll);
        for(auto &u : ans) u=tot,tot-=W[i++];
        return ans;
    }
    bool isLine = 1;
    for(int i = 0; i < N-1; i++)
        isLine &= (U[i]==i and V[i]==i+1);
    if(isLine){
        ll dp[N-1][2];
        dp[0][0] = dp[0][1] = 0;
        for(int i = 1; i < N; i++){
            dp[i][0] = min(dp[i-1][1],dp[i-1][0])+W[i-1];
            dp[i][1] = dp[i-1][0];
        }
        ans[1] = min(dp[N-1][0],dp[N-1][1]);
        return ans;
    }
    for(int i = 0; i < N-1; i++){
        adj[U[i]].pb({V[i],W[i]});
        adj[V[i]].pb({U[i],W[i]});
    }
    for(k = 1; k < n; k++)
        memset(dp,0,sizeof(dp)), dfs(0,-1), ans[k]=dp[0][0];
    return ans;
}

• Subtask 1: 5 / 5
• Subtask 2: 7 / 7
• Subtask 3: 0 / 14
• Subtask 4: 0 / 10
• Subtask 5: 0 / 17
• Subtask 6: 0 / 25
• Subtask 7: 0 / 22