[BZOJ2599][IOI2011]Race 树分治

题意

---

求树上权值和为K的最小简单路径

---

树分治

每次分治找到的Root，T[x]表示离Root距离为x的最小边数量，DFS时更新答案。

表示T数组开小竟然是TLE很不科学……

#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#define N 200010
#define M 1000010
#define inf 1<<30

using namespace std;

int n,k,G[N],cnt,root,mx[N],sz[N],V[N],tot,dis[N],d[N],Ans,t[M];
struct edge{
    int t,nx,w;
}E[N<<2];

inline char C(){
    return getchar();
    static char buf[100000],*p1=buf,*p2=buf;
    if(p1==p2){
        p2=(p1=buf)+fread(buf,1,100000,stdin);
        if(p1==p2)return EOF;
    }
    return *p1++;
}

inline void reaD(int &x){
    char Ch=C();x=0;
    for(;Ch>'9'||Ch<'0';Ch=C());
    for(;Ch>='0'&&Ch<='9';x=x*10+Ch-'0',Ch=C());
}

inline void InserT(int u,int v,int w){
    E[++cnt].t=v;E[cnt].nx=G[u];E[cnt].w=w;G[u]=cnt;
    E[++cnt].t=u;E[cnt].nx=G[v];E[cnt].w=w;G[v]=cnt;
}

inline int max(const int &a,const int &b){
    return a<b?b:a;
}

void GetRoot(int x,int f){
    sz[x]=1;mx[x]=0;
    for(int i=G[x];i;i=E[i].nx)
        if(E[i].t!=f&&!V[E[i].t]){
            GetRoot(E[i].t,x);
            sz[x]+=sz[E[i].t];
            mx[x]=max(mx[x],sz[E[i].t]);
        }
    mx[x]=max(mx[x],tot-sz[x]);
    if(mx[x]<mx[root]) root=x;
}

void calc(int x,int f){
    if(dis[x]<=k){Ans=min(Ans,d[x]+t[k-dis[x]]);}
    for(int i=G[x];i;i=E[i].nx)
        if(E[i].t!=f&&!V[E[i].t]){
            dis[E[i].t]=dis[x]+E[i].w;
            d[E[i].t]=d[x]+1;
            calc(E[i].t,x);
        }
}

void add(int x,int f){
    if(dis[x]<=k)t[dis[x]]=min(t[dis[x]],d[x]);
    for(int i=G[x];i;i=E[i].nx)
        if(E[i].t!=f&&!V[E[i].t]) add(E[i].t,x);
}

void desty(int x,int f){
    if(dis[x]<=k)t[dis[x]]=inf;
    for(int i=G[x];i;i=E[i].nx)
        if(E[i].t!=f&&!V[E[i].t]) desty(E[i].t,x);
}

void Solve(int x){
    V[x]=1;d[x]=0;t[0]=0;
    for(int i=G[x];i;i=E[i].nx)
        if(!V[E[i].t]){
            d[E[i].t]=d[x]+1;
            dis[E[i].t]=E[i].w;
            calc(E[i].t,x);
            add(E[i].t,x);
        }
    for(int i=G[x];i;i=E[i].nx)
        if(!V[E[i].t]) desty(E[i].t,x);
    for(int i=G[x];i;i=E[i].nx)
        if(!V[E[i].t]){
            root=0;tot=sz[E[i].t];
            GetRoot(E[i].t,0);
            Solve(root);
        }
}

int main(){
    #ifndef ONLINE_JUDGE
    freopen("1.in","r",stdin);
    freopen("1.out","w",stdout);
    #endif
    reaD(n);reaD(k);
    for(int i=1,u,v,w;i<n;i++)reaD(u),reaD(v),reaD(w),InserT(++u,++v,w);
    for(int i=1;i<=k;i++) t[i]=inf;
    mx[root=0]=inf;tot=n;Ans=inf;
    GetRoot(1,0);
    Solve(root);
    return printf("%d\n",Ans==inf?-1:Ans),0;
}