Submission #982637

#	Time	Username	Problem	Language	Result	Execution time	Memory
982637		vjudge1	Rainforest Jumps (APIO21_jumps)	C++17	Compilation error	0 ms	0 KiB

jumps

#include "roads.h"
#include <bits/stdc++.h>
#include <vector>
using namespace std;
vector<pair<long long, long long>> g[400001];
vector<long long> v;
long long n, i, j;
long long dp[400001][2];

void dfs(long long x, long long pr, long long k, long long c) {
	vector<long long> vv;
	for (auto y : g[x]) {
		if (y.first == pr) continue;
		dfs(y.first, x, k, y.second);
		dp[x][0] += dp[y.first][0];
	}
	dp[x][1] = dp[x][0];
	for (auto y : g[x]) {
		if (y.first == pr) continue;
		vv.push_back({-dp[y.first][0] + dp[y.first][1] + y.second});
	}
	sort(vv.begin(), vv.end());
	//long long w = vv.size() * 52;
	long long w = (long long)vv.size() - k;
	//for (long long i = 0; i < min((long long)vv.size(), w + 1); i++) {
	for (long long i = 0; i < vv.size(); i++) {
		if (i <= (long long)vv.size() - k) dp[x][0] += vv[i];
	}
	for (long long i = 0; i < vv.size(); i++) {
		if (i < (long long)vv.size() - k) dp[x][1] += vv[i];
	}
	//cout << x << " " << dp[x][0] << " " << dp[x][1] << "\n";
}

std::vector<long long> minimum_closure_costs(int N, std::vector<int> U, std::vector<int> V, std::vector<int> W) {
	n = N;
	long long s = 0;
	for (long long i = 0; i < N - 1; i++) {
		g[U[i]].push_back({V[i], W[i]});
		g[V[i]].push_back({U[i], W[i]});
	}
	for (long long i = 0; i < N; i++) {
		for (long long j = 0; j <= N; j++) dp[j][0] = dp[j][1] = 0;
		dfs(0, 0, i, 0);
		v.push_back(dp[0][1]);
	}
	reverse(v.begin(), v.end());
	assert(is_sorted(v.begin(), v.end()));
	reverse(v.begin(), v.end());
  return v;
}

/*
 *
5
0 1 1
0 2 4
0 3 3
2 4 2

 * */

Compilation message (stderr)

jumps.cpp:1:10: fatal error: roads.h: No such file or directory
    1 | #include "roads.h"
      |          ^~~~~~~~~
compilation terminated.