# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
725699 | sofija6 | 경주 (Race) (IOI11_race) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#include "race.h"
#define MAXN 200010
#define MAXK 1000010
using namespace std;
vector<pair<int,int> > G[MAXN];
bool check[MAXN];
int subtree[MAXN],k,res=INT_MAX,ans[MAXK];
void DFS_Subtree(int i,int p)
{
subtree[i]=1;
for (auto next : G[i])
{
if (next.first!=p && !check[next.first])
{
DFS_Subtree(next.first,i);
subtree[i]+=subtree[next.first];
}
}
}
int DFS_Centroid(int i,int p,int root)
{
for (auto next : G[i])
{
if (next.first!=p && !check[next.first] && subtree[next.first]>subtree[root]/2)
return DFS_Centroid(next.first,i,root);
}
return i;
}
void DFS_Paths(int i,int p,int d,int cnt)
{
if (d>k)
return;
if (ans[k-d])
res=min(res,ans[k-d]+cnt);
for (auto next : G[i])
{
if (next.first!=p)
DFS_Paths(next.first,i,d+next.second,cnt+1);
}
}
void DFS_Fill(int i,int p,int d,int cnt)
{
if (d>k)
return;
ans[d]=(!ans[d])?cnt : min(ans[d],cnt);
for (auto next : G[i])
{
if (next.first!=p)
DFS_Fill(next.first,i,d+next.second,cnt+1);
}
}
void Solve(int i)
{
DFS_Subtree(i,i);
int c=DFS_Centroid(i,i,i);
check[c]=true;
for (auto next : G[c])
{
DFS_Paths(next.first,c,next.second,1);
DFS_Fill(next.first,c,next.second,1);
}
for (auto next : G[c])
{
if (!check[next.first])
Solve(next.first);
}
}