【BZOJ2599】Race（点分治）

题面

BZOJ权限题，洛谷

题解

好久没写过点分治了。。。
在ppl的帮助下终于想起来了
orz ppl

首先回忆一下怎么求有没有正好是$K$的路径
维护一个表示距离的桶
对于当前重心，依次插入每棵子树的距离值
然后检查是否存在即可

显然加一步，求最短的路径数
那么，把原来的是否存在的01数组
改为记录最短路径数的一个$int$数组
每次插入的时候去取$min$就行了

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 220000
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
struct Line{int v,next,w;}e[MAX<<1];
int h[MAX],cnt=1;
inline void Add(int u,int v,int w){e[cnt]=(Line){v,h[u],w};h[u]=cnt++;}
int size[MAX],Size,K,n,dep[MAX],dis[MAX],ans;
bool vis[MAX];
int mx,root;
int ext[12000000];
int S[MAX],top;
int SS[MAX],Top;
void Getroot(int u,int ff)
{
    size[u]=1;
    int ret=0;
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;
        if(v==ff||vis[v])continue;
        Getroot(v,u);
        ret=max(ret,size[v]);
        size[u]+=size[v];
    }
    ret=max(Size-size[u],ret);
    if(ret<mx)mx=ret,root=u;
}
void Getdep(int u,int ff,int ss,int st)
{
    if(ss>K)return;
    SS[++Top]=S[++top]=u;
    dep[u]=ss;dis[u]=st;
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;
        if(v==ff||vis[v])continue;
        Getdep(v,u,ss+e[i].w,st+1);
    }
}
void Calc(int u,int ff,int W)
{
    top=0;Getdep(u,ff,W,1);
    for(int i=1;i<=top;++i)
    {
        int dd=dep[S[i]];
        if(!ext[K-dd])continue;
        ans=min(ans,dis[S[i]]+ext[K-dd]);
    }
    for(int i=1;i<=top;++i)
    {
        int dd=dep[S[i]];
        if(!ext[dd])ext[dd]=dis[S[i]];
        else ext[dd]=min(ext[dd],dis[S[i]]);
    }
}
void Work(int u)
{
    vis[u]=true;Top=0;
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;
        if(vis[v])continue;
        Calc(v,u,e[i].w);
    }
    if(ext[K])ans=min(ans,ext[K]);
    for(int i=1;i<=Top;++i)ext[dep[SS[i]]]=0;
    for(int i=h[u];i;i=e[i].next)
    {
        int v=e[i].v;
        if(vis[v])continue;
        Size=size[v],mx=n;
        Getroot(v,u);
        Work(root);
    }
}
int main()
{
    n=read();K=read();
    for(int i=1;i<n;++i)
    {
        int u=read()+1,v=read()+1,w=read();
        Add(u,v,w);Add(v,u,w);
    }
    Size=mx=n;Getroot(1,0);
    ans=1e9;
    Work(root);
    printf("%d\n",ans==1e9?-1:ans);
    return 0;
}