| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1035972 | andrewp | 경주 (Race) (IOI11_race) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "race.h"
#include <bits/stdc++.h>
using namespace std;
using pii = pair<int, int>;
using ll = int64_t;
#define int ll
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define dbg(x) cerr << #x << ": " << x << '\n';
const int N = 2e5 + 20, INF = 1e9 + 20;
int n, price[N], dep[N];
int k, ans = INF;
vector<pii> g[N];
map<int, int> mp[N];
void build(int x, int par, int pr, int d) {
price[x] = pr, dep[x] = d;
for (auto u : g[x]) {
if (u.first != par) {
build(u.first, x, pr + u.second, d + 1);
}
}
}
void dfs(int x, int par) {
mp[x][price[x]] = dep[x];
for (auto u : g[x]) {
if (u.first != par) {
dfs(u.first, x);
if (mp[u.first].size() > mp[x].size()) {
mp[x].swap(mp[u.first]);
}
for (auto add : mp[u.first]) {
int calc = 2 * price[x] + k - add.first;
if (mp[x].count(calc)) {
ans = min(ans, mp[x][calc] + add.second - 2 * dep[x]);
}
}
for (auto add : mp[u.first]) {
if (mp[x].count(add.first)) {
mp[x][add.first] = min(mp[x][add.first], add.second);
} else {
mp[x][add.first] = add.second;
}
}
if (mp[x].count(k + price[x])) {
ans = min(ans, mp[x][k + price[x]] - dep[x]);
}
}
}
// if (x == 1) dbg(mp[x].count(2));
}
int best_path(int NN, int K, int H[][2], int LEN[]) {
n = NN, k = K;
for (int i = 0; i < n - 1; i++) {
int u = H[i][0];
int v = H[i][1];
int w = LEN[i];
g[u].push_back({v, w});
g[v].push_back({u, w});
}
dep[0] = 0, price[0] = 0;
build(0, -1, 0, 0);
// for (int i = 0; i < n; i++) cerr << price[i] << ' ';
// cout << '\n';
dfs(0, -1);
return ans;
}
/*int32_t main() {
ios::sync_with_stdio(false); cin.tie(nullptr);
cout.tie(nullptr); cerr.tie(nullptr);
int NN, K;
cin >> NN >> K;
int H[NN][2], LEN[NN];
for (int i = 0; i < NN - 1; i++) {
cin >> H[i][0] >> H[i][1];
}
for (int i = 0; i < NN - 1; i++) {
cin >> LEN[i];
}
int ans = best_path(NN, K, H, LEN);
cout << "Solution: " << (ans == INF ? -1 : ans) << '\n';
}*/