Submission #93998

#TimeUsernameProblemLanguageResultExecution timeMemory
93998fjzzq2002Race (IOI11_race)C++14
100 / 100
1066 ms52856 KiB
#include "race.h"
#include <bits/stdc++.h>
using namespace std;
namespace Wrap
{
#define Edg int M=0,fst[SZ],vb[SZ],nxt[SZ];void ad_de(int a,int b){++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b;}void adde(int a,int b){ad_de(a,b);ad_de(b,a);}
#define Edgc int M=0,fst[SZ],vb[SZ],nxt[SZ],vc[SZ];void ad_de(int a,int b,int c){++M;nxt[M]=fst[a];fst[a]=M;vb[M]=b;vc[M]=c;}void adde(int a,int b,int c){ad_de(a,b,c);ad_de(b,a,c);}
#define es(x,e) (int e=fst[x];e;e=nxt[e])
#define esb(x,e,b) (int e=fst[x],b=vb[e];e;e=nxt[e],b=vb[e])
#define SZ 666666
typedef long long ll;
Edgc
int n,K,mi=2e9;
int sz[SZ],son[SZ],dep[SZ];
ll d1[SZ];
void dfs(int x,int f=SZ-1)
{
	dep[x]=dep[f]+1;
	sz[x]=1; son[x]=SZ-1;
	for esb(x,e,b) if(b!=f)
	{
		d1[b]=d1[x]+vc[e]; dfs(b,x); sz[x]+=sz[b];
		if(sz[b]>sz[son[x]]) son[x]=b;
	}
}
map<ll,int> ms;
int&gp(ll x)
{
	if(!ms.count(x))
		return ms[x]=2e9;
	return ms[x];
}
ll L; int ca;
void qry_pnt(int g)
{
	//d1[g]+d1[?]-L=K
	ca=min(ca,gp(K+L-d1[g])+dep[g]);
}
void add_pnt(int g)
{
	auto&t=gp(d1[g]);
	t=min(t,dep[g]);
}
template<class T>
void go(int x,int f,T*fun)
{
	fun(x);
	for esb(x,e,b) if(b!=f) go(b,x,fun);
}
void dfs2(int x,int t,bool k=1,int f=SZ-1)
{
	for esb(x,e,b) if(b!=f&&b!=son[x]) dfs2(b,b,0,x);
	if(son[x]<=n) dfs2(son[x],t,1,x);
	L=d1[x]*2; ca=2e9;
	for esb(x,e,b) if(b!=f&&b!=son[x])
		go(b,x,qry_pnt),go(b,x,add_pnt);
	qry_pnt(x); add_pnt(x);
	mi=min(mi,ca-dep[x]*2);
	if(!k) ms.clear();
}
}
int best_path(int N, int K_, int H[][2], int L[])
{
	using namespace Wrap;
	K=K_; n=N; mi=2e9;
	for(int i=0;i<N-1;++i)
		adde(H[i][0],H[i][1],L[i]);
	dfs(0); dfs2(0,0);
	return (mi>1e9)?(-1):mi;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...