제출 #339528

#제출 시각아이디문제언어결과실행 시간메모리
339528penguinhacker경주 (Race) (IOI11_race)C++14
100 / 100
1501 ms42860 KiB
#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define ar array

const int mxN = 200000;
int n, k, ans, sz[mxN];
bool dead[mxN];
vector<pair<int, int>> adj[mxN];

int dfs_sz(int u, int p) {
	sz[u] = 1;
	for (pair<int, int> v : adj[u]) if (v.first != p && !dead[v.first])
		sz[u] += dfs_sz(v.first, u);
	return sz[u];
}

int get_centroid(int u, int p, int tree_sz) {
	for (pair<int, int> v : adj[u]) if (v.first != p && !dead[v.first] && 2 * sz[v.first] > tree_sz)
		return get_centroid(v.first, u, tree_sz);
	return u;
}

map<int, int> mp, mp2;
void upd(int wd, int d, map<int, int>& m) {
	auto it = m.find(wd);
	if (it == m.end()) {
		m[wd] = d;
	}
	else {
		it->second = min(it->second, d);
	}
}

void dfs(int u, int p, int d, int wd, map<int, int>& m) {
	if (wd > k) return;
	if (wd == k) {
		ans = min(ans, d);
		return;
	}
	upd(wd, d, m);
	for (pair<int, int> v : adj[u]) if (v.first != p && !dead[v.first]) {
		dfs(v.first, u, d + 1, wd + v.second, m);
	}
}

void centroid(int u = 0) {
	u = get_centroid(u, -1, dfs_sz(u, -1));
	dfs_sz(u, -1);
	int big = -1;
	for (pair<int, int> v : adj[u]) if (!dead[v.first]) {
		if (big == -1 || sz[v.first] > sz[big]) {
			big = v.first;
		}
	}
	if (big == -1) return;
	for (pair<int, int> v : adj[u]) if (v.first == big) {
		dfs(v.first, u, 1, v.second, mp);
	}
	for (pair<int, int> v : adj[u]) if (!dead[v.first] && v.first != big) {
		dfs(v.first, u, 1, v.second, mp2);
		for (const pair<int, int>& p : mp2) {
			auto it = mp.find(k - p.first);
			if (it != mp.end()) {
				ans = min(ans, p.second + it->second);
			}
		}
		for (const pair<int, int>& p : mp2) {
			upd(p.first, p.second, mp);
		}
		mp2.clear();
	}	
	mp.clear();
	dead[u] = 1;
	for (pair<int, int> v : adj[u]) if (!dead[v.first]) {
		centroid(v.first);
	}
	dead[u] = 0;
}

int best_path(int _n, int _k, int h[][2], int l[]) {
	n = _n, k = _k, ans = INT_MAX;
	for (int i = 0; i < n - 1; ++i) {
		int u = h[i][0], v = h[i][1], w = l[i];
		adj[u].emplace_back(v, w);
		adj[v].emplace_back(u, w);
	}
	centroid();
	if (ans == INT_MAX) ans = -1;
	return ans;
}

/*int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	int n, k; cin >> n >> k;
	int h[n - 1][2];
	int l[n - 1];
	for (int i = 0; i < n - 1; ++i) {
		cin >> h[i][0] >> h[i][1];
	}
	for (int i = 0; i < n - 1; ++i) {
		cin >> l[i];
	}
	cout << best_path(n, k, h, l);
	return 0;
}
*/
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...