Submission #1223415

#TimeUsernameProblemLanguageResultExecution timeMemory
1223415SpyrosAlivRoad Closures (APIO21_roads)C++20
Compilation error
0 ms0 KiB
#include "roads.h" #include <bits/stdc++.h> using namespace std; #define ll long long const ll INF = 1e17; vector<vector<pair<int, ll>>> tree; vector<vector<ll>> dp; int n; void dfs(int node, int par = 0) { int ch = 0; dp[node][0] = 0; for (auto [next, w]: tree[node]) { if (next == par) continue; dfs(next, node); dp[node][0] += w + dp[next][0]; } for (int k = 1; k <= n; k++) { ll ans = 0; vector<ll> ops; for (auto [next, w]: tree[node]) { if (next == par) continue; ops.push_back(dp[next][k-1] - (dp[next][k] + w)); ans += dp[next][k] + w; } sort(ops.begin(), ops.end()); if (ops.size() < k) { dp[node][k] = dp[node][k-1]; continue; } for (int j = 0; j < k; j++) { ans += ops[j]; } dp[node][k] = min(ans, dp[node][k-1]); } } vector<ll> minimum_closure_costs(int N, vector<int> U, vector<int> V, vector<ll> W) { n = N; tree.clear(); tree.resize(n+1); for (int i = 0; i < n-1; i++) { U[i]++; V[i]++; tree[U[i]].push_back({V[i], W[i]}); tree[V[i]].push_back({U[i], W[i]}); } dp.assign(n+1, vector<ll>(n+1, INF)); dfs(1); vector<ll> ans; for (int k = 1; k < n; k++) dp[1][k] = min(dp[1][k], dp[1][k-1]); for (int k = 0; k < n; k++) ans.push_back(dp[1][k]); return ans; }

Compilation message (stderr)

/usr/bin/ld: /tmp/ccPB29w7.o: in function `main':
grader.cpp:(.text.startup+0x278): undefined reference to `minimum_closure_costs(int, std::vector<int, std::allocator<int> >, std::vector<int, std::allocator<int> >, std::vector<int, std::allocator<int> >)'
collect2: error: ld returned 1 exit status