Submission #677585

#	Time	Username	Problem	Language	Result	Execution time	Memory
677585		YENGOYAN	Road Closures (APIO21_roads)	C++17	0 / 100	381 ms	1048576 KiB

roads

#include "roads.h"
#include <iostream>
#include <vector>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_map>
#include <cmath>
#include <climits>
#include <algorithm>
#include <random>
#include <queue>
#include <deque>
#include <iomanip>
#include <string>
#include <tuple>
#include <bitset>
#include <chrono>
#include <ctime>
#include <fstream>
#include <stack>
#include <cstdio>
using namespace std;
using ll = long long;

vector<vector<pair<int, int>>> gp;
vector<vector<ll>> dp;

void dfs(int u, int p, int k) {
	vector<ll> vec;
	for (pair<int, int>& v : gp[u]) {
		if (v.first == p) continue;
		dfs(v.first, u, k);
		dp[u][k] += dp[v.first][k];
		vec.push_back(max(0ll, dp[v.first][k - 1] + v.second - dp[v.first][k]));
	}
	//dp[u][0] =
	sort(vec.rbegin(), vec.rend());
	for (int i = 0; i < min(int(vec.size()), k); ++i) dp[u][k] += vec[i];
}


vector<ll> minimum_closure_costs(int n, vector<int> U,vector<int> V,vector<int> W) {
	gp = vector<vector<pair<int, int>>>(n);
	dp = vector<vector<ll>>(n, vector<ll>(n));
	ll sm = 0;
	for (int i = 0; i < n - 1; ++i) {
		gp[U[i]].push_back({ V[i], W[i] });
		gp[V[i]].push_back({ U[i], W[i] });
		sm += W[i];
	}
	vector<ll> vec(n);
	vec[0] = sm;
	for (int i = 1; i < n; ++i) {
		dfs(0, -1, i);
		vec[i] = sm - dp[0][i];
	}
	return vec;
}

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