BZOJ 2599: [IOI2011]Race 点分治

记录每一次分治时，达到重心的dep，加到一个桶中，没添加一个节点，查询桶中是否存在，之后加和和答案取最小值

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <queue>
#include <iostream>
#include <cstdlib>
using namespace std;
#define N 300505
#define ll long long
struct node
{
int to,next,val;
}e[N<<1];
int siz
,dep
,ans=1<<30,v[N*5],rot,sum,vis
,mx
,head
,cnt,K,n;
ll dis
;
void add(int x,int y,int z)
{
e[cnt].to=y;
e[cnt].next=head[x];
e[cnt].val=z;
head[x]=cnt++;
return ;
}
void get_root(int x,int from)
{
siz[x]=1;mx[x]=0;
for(int i=head[x];i!=-1;i=e[i].next)
{
int to1=e[i].to;
if(!vis[to1]&&to1!=from)
{
get_root(to1,x);
siz[x]+=siz[to1];
mx[x]=max(siz[to1],mx[x]);
}
}
mx[x]=max(mx[x],sum-siz[x]);
if(mx[x]<mx[rot])rot=x;
}
void insert(int x,int from)
{
if(dis[x]<=K)v[dis[x]]=min(v[dis[x]],dep[x]);
for(int i=head[x];i!=-1;i=e[i].next)
{
int to1=e[i].to;
if(!vis[to1]&&to1!=from)
{
insert(to1,x);
}
}
}
void clear(int x,int from)
{
if(dis[x]<=K)v[dis[x]]=1<<30;
for(int i=head[x];i!=-1;i=e[i].next)
{
int to1=e[i].to;
if(!vis[to1]&&to1!=from)
{
clear(to1,x);
}
}
}
void calc(int x,int from)
{
if(dis[x]<=K)ans=min(ans,dep[x]+v[K-dis[x]]);
for(int i=head[x];i!=-1;i=e[i].next)
{
int to1=e[i].to;
if(to1!=from&&!vis[to1])
{
dep[to1]=dep[x]+1;
dis[to1]=dis[x]+e[i].val;
calc(to1,x);
}
}
}
void dfs(int x)
{
vis[x]=1;v[0]=0;
for(int i=head[x];i!=-1;i=e[i].next)
{
int to1=e[i].to;
if(!vis[to1])
{
dep[to1]=1,dis[to1]=e[i].val;
calc(to1,0);
insert(to1,0);
}
}
for(int i=head[x];i!=-1;i=e[i].next)
{
int to1=e[i].to;
if(!vis[to1])
{
clear(to1,0);
}
}
for(int i=head[x];i!=-1;i=e[i].next)
{
int to1=e[i].to;
if(!vis[to1])
{
sum=siz[to1];rot=0;
get_root(to1,0);
dfs(rot);
}
}
}
int main()
{
memset(head,-1,sizeof(head));
scanf("%d%d",&n,&K);
for(int i=1;i<n;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
add(x+1,y+1,z);
add(y+1,x+1,z);
}
for(int i=1;i<=K;i++)v[i]=1<<30;
mx[0]=sum=n;
get_root(1,0);
dfs(rot);
if(ans==1<<30)
{
puts("-1");
return 0;
}
printf("%d\n",ans);
return 0;
}