제출 #1281037

#제출 시각아이디문제언어결과실행 시간메모리
1281037trinm01경주 (Race) (IOI11_race)C++20
0 / 100
1 ms332 KiB
#include "race.h"
#include <bits/stdc++.h>
using namespace std;

//#define int long long 
#define ll long long
#define FOR(i, l, r) for (int i = (l); i <= (r); i++)
#define FOD(i, r, l) for (int i = (r); i >= (l); i--)
#define fi first
#define se second
#define pii pair<int, int>

const ll mod = 1e9 + 7;
const int MAXN = 2e5 + 5;
const ll oo = 1e9 + 7;  
const int base = 10;

int n, k;
vector<pii> adj[MAXN];

int del[MAXN], sz[MAXN];
void count(int u, int p){
	sz[u]=1;
	for(auto [v, w]:adj[u]){
		if(v==p || del[v]) continue;
		count(v, u);
		sz[u]+=sz[v];
	}
}
int find(int u, int p, int m){
	for(auto [v, w]:adj[u]){
		if(v==p || del[v]) continue;
		if(sz[v]>m/2){
			return find(v, u, m);
		}
	}
	return u;
}
int ans=oo;
int d[MAXN], h[MAXN];
int mp[1000005];
void reset(int u, int p){
	if(d[u]<=1e6){
		mp[d[u]]=oo;
	}
	if(k-d[u]>=0){
		mp[k-d[u]]=oo;
	}
	for(auto [v, w]:adj[u]){
		if(v==p || del[v]) continue;
		h[v]=h[u]+1;
		d[v]=d[u]+w;
		reset(v, u);
	}
}
void dfs1(int u, int p){
	if(k-d[u]>=0){
		ans=min(ans, h[u]+mp[k-d[u]]);
	}
	for(auto [v, w]:adj[u]){
		if(v==p || del[v]) continue;
		h[v]=h[u]+1;
		d[v]=d[u]+w;
		dfs1(v, u);
	}
}
void dfs2(int u, int p){
	if(d[u]<=1e6)
		mp[d[u]]=min(mp[d[u]], h[u]);
	for(auto [v, w]:adj[u]){
		if(v==p || del[v]) continue;
		dfs2(v, u);
	}
}
void solve(int u){
	h[u]=d[u]=0;
	reset(u, 0);
	mp[0]=0;
	for(auto [v, w]:adj[u]){
		if(del[v]) continue;
		h[v]=h[u]+1;
		d[v]=d[u]+w;
		dfs1(v, u);
		dfs2(v, u);
	}
	
	
	reset(u, 0);
	mp[0]=0;
	FOD(i, (int)adj[u].size()-1, 0){
		auto [v, w]=adj[u][i];
		if(del[v]) continue;
		h[v]=h[u]+1;
		d[v]=d[u]+w;
		dfs1(v, u);
		dfs2(v, u);
	}
}
void centroid(int u, int p){
	count(u, 0);
	int root=find(u, 0, sz[u]);
	
	solve(root);
	
	del[root]=1;
	for(auto [v, w]:adj[root]){
		if(del[v]) continue;
		centroid(v, root);
	}
}

int best_path(int N, int K, int H[][2], int L[])
{
	n=N, k=K;
	FOR(i, 0, n-2){
		int u, v, c;
		u=H[i][0];
		v=H[i][1];
		c=L[i];
		u++;
		v++;
		adj[u].push_back({v, c});
    	adj[v].push_back({u, c});
	}
	
	centroid(1, 0);
    
    if(ans==oo) ans=-1;
    cout << ans;
    
    return ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...