点分治 [IOI2011]Race

BZOJ2599.

放板子。利用agc009-D 代码简洁了很多

#include <bits/stdc++.h>
#define N 200005
using namespace std;
int n,m,K,F[N][30],H[N],x,y,z,h[N],f[1000005],T,O,e[N],ans,d[N],g[N],t;
vector <int> a[N],b[N];
void QH(int x,int fa){
    int y,z=-1;
    for (int i=0;i<a[x].size();++i)
    if (a[x][i]!=fa){
        QH(y=a[x][i],x);
        for (int j=0;j<30;++j){
            if (F[y][j]&F[x][j]) z=max(z,j);
            F[x][j]|=F[y][j];
        }
    }
    for (++z;F[x][z];++z);
    F[x][z]=1; H[x]=z;
    for (--z;~z;--z) F[x][z]=0;
}
void dfs(int x,int y,int o){
    if (h[x]>K) return;
    d[++t]=h[x]; g[t]=o; e[++T]=x;
    ans=min(ans,f[K-h[x]]+o);
    for (int i=0;i<a[x].size();++i)
    if (H[a[x][i]]<O&&a[x][i]!=y){
        h[a[x][i]]=h[x]+b[x][i];
        dfs(a[x][i],x,o+1);
    }
}
void go(int x){
    O=H[x]; h[x]=T=0;
    for (int i=0;i<a[x].size();++i)
    if (H[a[x][i]]<O){
        h[a[x][i]]=h[x]+b[x][i];
        t=0; dfs(a[x][i],x,1);
        for (int j=1;j<=t;++j) f[d[j]]=min(f[d[j]],g[j]);
    }
    for (int i=1;i<=T;++i) f[h[e[i]]]=n+5;
}
int main(){
    scanf("%d%d",&n,&K);
    for (int i=1;i<n;++i){
        scanf("%d%d%d",&x,&y,&z);
        ++x; ++y;
        a[x].push_back(y);
        b[x].push_back(z);
        a[y].push_back(x);
        b[y].push_back(z);
    }
    QH(1,0); ans=n+5;
    for (int i=1;i<=K;++i) f[i]=n+5;
    for (int i=1;i<=n;++i) go(i);
    printf("%d
",ans>n?-1:ans);
    return 0;
}