# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
572552 | vladislav11 | 경주 (Race) (IOI11_race) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
int n, ans;
ll k;
vector< vector< pair<int,int> > > grp;
vector<int> ssize, h;
vector<ll> pref;
int dfs_prec ( int v, int p )
{
for ( auto& [to,w] : grp[v] )
if ( to != p )
{
h[to] = h[v] + 1;
pref[to] = pref[v] + w;
ssize[v] += dfs_prec( to, v );
}
return ssize[v];
}
vector< vector<int>* > ver;
map<ll,int> mini;
void upd ( int in, int v )
{
if ( pref[in] + k == pref[v] )
ans = min( ans, h[v] - h[in] );
else if ( pref[v] - pref[in] < k && mini.count( pref[in] + k - (pref[v] - pref[in]) ) )
ans = min( ans, ( h[v] - h[in] ) + ( mini[ pref[in] + k - (pref[v] - pref[in]) ] - h[in] ) );
}
void add ( int v )
{
if ( mini.count( pref[v] ) )
mini[ pref[v] ] = min( mini[ pref[v] ], h[v] );
else
mini[ pref[v] ] = h[v];
}
void dfs ( int v, int p, bool keep )
{
//cout << "dfs: " << v << ' ';
int bigs = -1, bigv = -1;
for ( auto& [to,w] : grp[v] )
if ( to != p && ssize[to] > bigs )
{
bigs = ssize[to];
bigv = to;
}
//cout << bigs << ' ' << bigv << '\n';
for ( auto& [to,w] : grp[v] )
if ( to != p && to != bigv )
dfs( to, v, false );
if ( bigv == -1 )
ver[v] = new vector<int>();
else
{
dfs( bigv, v, true );
ver[v] = ver[bigv];
}
ver[v]->push_back( v );
for ( auto& [to,w] : grp[v] )
if ( to != p && to != bigv )
{
for ( auto& el : *ver[to] )
upd( v, el );
for ( auto& el : *ver[to] )
{
ver[v]->push_back( el );
add( el );
}
}
if ( mini.count( pref[v] + k ) )
ans = min( ans, mini[ pref[v] + k ] - h[v] );
mini[ pref[v] ] = h[v];
/*cout << "mini " << v << ": ";
for ( auto& [x,y] : mini )
cout << "(" << x << ' ' << y << ") ";
cout << '\n';*/
if ( !keep )
mini.clear();
}
int best_path ( int n_, int k_, vector< pair<int,int> > h_, vector<int> l_ )
{
n = n_;
k = k_;
grp.assign( n, vector< pair<int,int> >() );
for ( int i=0; i<n-1; i++ )
{
grp[ h_[i].first ].push_back({ h_[i].second, l_[i] });
grp[ h_[i].second ].push_back({ h_[i].first, l_[i] });
}
ssize.assign( n, 1 );
h.assign( n, 1 );
pref.assign( n, 0 );
dfs_prec( 0, -1 );
ver.assign( n, nullptr );
mini.clear();
ans = 2e9;
dfs( 0, -1, false );
return ans == 2e9 ? -1 : ans;
}