| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 784665 | vjudge1 | 경주 (Race) (IOI11_race) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "race.h"
using namespace std;
#define endl '\n'
#define ll long long
#define all(x) x.begin(), x.end()
vector<pair<int, int>> g[200005];
bool processed[200005];
int sz[200005];
int getSize(int cur, int par = -1) {
sz[cur] = 1;
for (auto next : g[cur]) {
if (next.first != par && !processed[next.first]) sz[cur] += getSize(next.first, cur);
}
return sz[cur];
}
int getCentroid(int cur, int des, int par = -1) {
for (auto next : g[cur]) {
if (next.first != par && !processed[cur] && sz[cur] > des / 2) return getCentroid(next.first, des, cur);
}
return cur;
}
int ans = 10e7, k;
map<int, int> cnt;
set<int> s;
void solve(int cur, int par, int depth, int sum, bool t) {
if (sum > k || depth >= ans) return;
if (sum == k) {
ans = depth;
return;
}
s.insert(sum);
if (!t) if (cnt.count(k - sum)) ans = min(ans, depth + cnt[k - sum]);
else cnt[sum] = min(cnt[sum], depth);
for (auto next : g[cur]) {
if (next.first != par && !processed[next.first] && sum + next.second <= k) {
solve(next.first, cur, depth + 1, sum + next.second, t);
}
}
}
void decompose(int cur = 0) {
getSize(cur);
cur = getCentroid(cur, getSize(cur));
processed[cur] = 1;
for (auto next : g[cur]) {
if (!processed[next.first] && next.second <= k) {
solve(cur, -1, 1, next.second, 0);
solve(cur, -1, 1, next.second, 1);
}
}
for (auto x : s) cnt.clear(x);
s.clear();
for (auto next : g[cur]) {
if (!processed[next.first]) decompose(next.first);
}
}
int best_path(int N, int K, int H[][2], int L[]) {
k = K;
for (int i = 0; i < N - 1; i++) {
g[H[i][0]].push_back({H[i][1], L[i]});
g[H[i][1]].push_back({H[i][0], L[i]});
}
decompose();
if (ans == 10e7) return -1;
else return ans;
}