| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 302287 | nhdtxdy | 경주 (Race) (IOI11_race) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#define ll long long
#define int long long
#define pb push_back
#define eb emplace_back
#define Size(v) (int)v.size()
using namespace std;
// Always
const int nax = 2e5 + 5, INF = 1e9 + 7;
int n, k;
vector<pair<int, int>> adj[nax];
map<int, int> vec[nax];
int dist[nax], ans[nax], sz[nax], dep[nax];
void getsz(int v, int p = 0) {
sz[v] = 1;
for (auto &q : adj[v]) {
int c = q.first, w = q.second;
if (c == p) continue;
dep[c] = dep[v] + 1;
dist[c] = dist[v] + w;
getsz(c, v);
sz[v] += sz[c];
}
}
void dfs(int v, int p = 0, bool keep = false) {
int maxi = -1, bigchild = -1;
for (auto &q : adj[v]) {
int u = q.first;
if (u == p) continue;
if (maxi < sz[u]) {
maxi = sz[u];
bigchild = u;
}
}
for (auto &q : adj[v]) {
int u = q.first;
if (u == p || u == bigchild) continue;
dfs(u, v, false);
}
if (bigchild != -1) {
dfs(bigchild, v, true);
swap(vec[v], vec[bigchild]);
}
vec[v][dist[v]] = dep[v];
int subtract = dist[v];
if (vec[v].find(k + dist[v]) != vec[v].end()) {
ans[v] = min(ans[v], vec[v][k + dist[v]] - dep[v]);
}
int target = k + 2 * subtract;
for (auto &q : adj[v]) {
int c = q.first;
if (c == p || c == bigchild) continue;
if (vec[c].find(k + dist[v]) != vec[c].end()) {
ans[v] = min(ans[v], vec[c][k + dist[v]] - dep[v]);
}
for (auto &u : vec[c]) {
if (vec[v].find(target - u.first) != vec[v].end()) {
ans[v] = min(ans[v], u.second + vec[v][target - u.first] - 2 * dep[v]);
}
}
for (auto &u : vec[c]) {
if (vec[v].find(u.first) != vec[v].end()) vec[v][u.first] = min(vec[v][u.first], u.second);
else vec[v][u.first] = u.second;
}
}
}
int32_t best_path(int32_t N, int32_t K, int32_t H[][2], int32_t L[]) {
fill(ans, ans + nax, INF);
n = N;
k = K;
for (int i = 0; i < n - 1; ++i) {
int u = H[i][0], v = H[i][1];
++u, ++v;
adj[u].eb(v, L[i]);
adj[v].eb(u, L[i]);
}
getsz(1);
dfs(1);
int32_t res = INF;
for (int i = 1; i <= n; ++i) res = min(res, ans[i]);
return res;
}