| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 833848 | algorithm16 | 경주 (Race) (IOI11_race) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<iostream>
#include<vector>
#include<set>
#include<algorithm>
#include "race.h"
using namespace std;
typedef long long int llint;
vector <pair<int,llint> > v[200005];
set <pair<llint,llint> > s[200005];
llint k1,mn=1e9,h[25][2],l[25];
void f(int x,int y,llint dd,llint ll) {
if(s[x].size()<s[y].size()) swap(s[x],s[y]);
for(set <pair<llint,llint> >::iterator it=s[y].begin();it!=s[y].end();it++) {
llint l=(*it).first,d=(*it).second;
l-=ll;
d-=dd;
set <pair<llint,llint> >::iterator it2=s[x].lower_bound(make_pair(k1-l+ll,-1LL));
if(it2!=s[x].end() && (*it2).first==k1-l+ll) mn=min(mn,d+(*it2).second-dd);
}
for(set <pair<llint,llint> >::iterator it=s[y].begin();it!=s[y].end();) {
llint l=(*it).first,d=(*it).second;
l-=ll;
d-=dd;
it2=s[x].lower_bound(make_pair(l+ll,-1LL));
if(it2==s[x].end() || (*it2).first!=l+ll) s[x].insert((*it));
else if((*it2).second>(*it).second) {
s[x].erase(it2);
s[x].insert((*it));
}
it=s[y].erase(it);
}
}
void dfs(int x,int p,llint d,llint l) {
s[x].insert(make_pair(l,d));
for(int i=0;i<v[x].size();i++) {
if(v[x][i].first==p) continue;
dfs(v[x][i].first,x,d+1,l+v[x][i].second);
f(x,v[x][i].first,d,l);
}
}
int best_path(int n,int k,int h[][2],int l[])
{
if(!k) return 0;
k1=k;
for(int i=0;i<n-1;i++) {
v[h[i][0]].push_back(make_pair(h[i][1],l[i]));
v[h[i][1]].push_back(make_pair(h[i][0],l[i]));
}
dfs(0,0,0,0);
if(mn==1e9) mn=-1;
return mn;
}
/*int main()
{
int n,k;
cin >> n >> k;
for(int i=0;i<n-1;i++) {
int a,b,l1;
cin >> a >> b >> l1;
h[i][0]=a;
h[i][1]=b;
l[i]=l1;
}
cout << best_path(n,k);
}
*/