| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 865981 | yeediot | 경주 (Race) (IOI11_race) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
#define F first
#define S second
#define all(x) x.begin(),x.end()
#define pii pair<long long,long long>
#define pb push_back
#define sz(x) (long long)(x.size())
#define chmin(x,y) x=min(x,y)
#define chmax(x,y) x=max(x,y)
#define vi vector<long long>
#define vp vector<pii>
#define vvi vector<vi>
//Don't open the standings during contests.
const long long mxn=2e5+5;
vector<pii>adj[mxn];
long long cnt[mxn];
bool vis[mxn];
void cnt_sz(long long v,long long pa){
cnt[v]=1;
for(auto [u,d]:adj[v]){
if(vis[u] or u==pa)continue;
cnt_sz(u,v);
cnt[v]+=cnt[u];
}
}
long long find(long long v,long long pa,long long tar){
for(auto [u,d]:adj[v]){
if(vis[u] or u==pa)continue;
if(cnt[u]*2>tar)return find(u,v,tar);
}
return v;
}
long long ans=8e12;
vector<pii>dis;
void dfs(long long v,long long pa,long long l,long long len,long long k){
//cout<<v<<' '<<l<<' '<<len<<'\n';
if(l>k)return;
dis.pb({l,len});
for(auto [u,d]:adj[v]){
if(vis[u] or u==pa)continue;
dfs(u,v,l+1,len+d,k);
}
}
long long f[mxn];
void cd(long long v,long long k,long long n){
cnt_sz(v,v);
v=find(v,v,cnt[v]);
vis[v]=1;
vector<long long>undo;
for(auto [u,d]:adj[v]){
if(vis[u])continue;
dis.clear();
dfs(u,v,1,d,k);
for(long long j=0;j<sz(dis);j++){
chmin(ans,f[k-dis[j].S]+dis[j].F);
// cout<<u<<' '<<dis[j].S<<' '<<dis[j].F<<' '<<f[k-dis[j].S]<<' '<<ans<<'\n';
}
for(long long j=0;j<sz(dis);j++){
undo.pb(dis[j].S);
chmin(f[dis[j].S],dis[j].F);
}
}
for(auto x:undo)f[x]=8e12;
for(auto [u,d]:adj[v]){
if(vis[u])continue;
cd(u,k,n);
}
}
int best_path(int N,int K,int H[][2],int L[]){
for(long long i=0;i<N-1;i++){
adj[H[i][0]].pb({H[i][1],L[i]});
adj[H[i][1]].pb({H[i][0],L[i]});
}
for(long long i=0;i<=K;i++){
f[i]=8e12;
}
cd(0,K,N);
return (ans==8e12?-1:ans);
}
signed main(){
ios::sync_with_stdio(0);
cin.tie(0);cout.tie(0);
int n,k;
cin>>n>>k;
int h[n-1][2];
int l[n-1];
for(long long i=0;i<n-1;i++){
cin>>h[i][0]>>h[i][1];
}
for(long long i=0;i<n-1;i++){
cin>>l[i];
}
cout<<best_path(n,k,h,l);
}
/*
input:
11 12
0 1
0 2
2 3
3 4
4 5
0 6
6 7
6 8
8 9
8 10
3
4
5
4
6
3
2
5
6
7
*/