# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
413455 | souvenir_vayne | 공장들 (JOI14_factories) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "factories.h"
#define pb push_back
#define LINF 0x3f3f3f3f3f3f3f3f
#define endl '\n'
#define ll long long
#define f first
#define s second
using namespace std;
typedef pair<ll, ll> pii;
const int MAXN = 5e5+5;
ll sz[MAXN], used[MAXN], best[MAXN];
vector<pii> g[MAXN], ng[MAXN];
int getsz(int u, int p) {
sz[u] = 1;
for(pii &i : g[u])
if(i.f != p && !used[i.f])
sz[u] += getsz(i.f, u);
return sz[u];
}
int findc(int u, int p, int n) {
for(pii &i : g[u])
if(i.f != p && !used[i.f] && sz[i.f] > n/2)
return findc(i.f, u, n);
return u;
}
void getng(int u, int p, int c, ll dist) {
ng[u].pb({c, dist});
for(pii &i : g[u])
if(i.f != p && !used[i.f])
getng(i.f, u, c, dist + i.s);
}
void build(int u) {
int c = findc(u, -1, getsz(u, -1));
getng(c, -1, c, 0);
used[c] = 1;
for(pii &i : g[c])
if(!used[i.f])
build(i.f);
}
void Init(int n, int a[], int b[], int c[]) {
for(int i = 0; i < n-1; i++) {
g[a[i]].pb({b[i], c[i]});
g[b[i]].pb({a[i], c[i]});
}
build(0);
fill(best, best + N, LINF);
}
ll Query(int s, int x[], int t, int y[]) {
ll ans = LINF;
for(int u = 0; u < s; u++)
for(pii &i : ng[x[u]])
best[i.f] = min(best[i.f], i.s);
for(int u = 0; u < t; u++)
for(pii &i : ng[y[u]])
ans = min(ans, best[i.f] + i.s);
for(int u = 0; u < s; u++)
for(pii &i : ng[x[u]])
best[i.f] = LINF;
return ans;
}