【bzoj 2599】Race

傳送門~

解題思路

點分治練習。
注意k值太大不能memset，所以用了時間戳。
注意不能走的點的標記的處理。
代碼：

#include<algorithm>
#include<iostream>
#include<cmath>
#include<cstring>
#include<string>
#include<cstdio>
#include<cstdlib>
using namespace std;
const int mx=2147483640;
int hed[200005],nex[400005],lb[400005],val[400005];
int f[1000005],tcl[1000005];
int siz[200005],pa[200005];
int n,k,lo,T,ans=mx,u,sum;
inline void update(int x){
    if(tcl[x]==T) return ;
    f[x]=mx;tcl[x]=T;
}
void add(int x,int y,int num){
    lo++;
    nex[lo]=hed[x];
    hed[x]=lo;
    lb[lo]=y;
    val[lo]=num;
}
int get_siz(int x,int fa){
    int r=1;
    for(int i=hed[x];i!=0;i=nex[i]){
        if(lb[i]!=fa && !pa[lb[i]])
        r+=get_siz(lb[i],x);
    }
    return r;
}
int getu(int x,int fa){
    int r=1,flag=0;
    for(int i=hed[x];i!=0;i=nex[i])
    if(lb[i]!=fa && !pa[lb[i]]){
        int r0=getu(lb[i],x);
        if(r0*2>sum) flag=1;
        r+=r0;
    }
    if((sum-r)*2>sum) flag=1;
    if(!flag) u=x;
    return r;
}

void dfs1(int x,int fa,int num,int pnum){
    if(num>k) return ;update(k-num);
    if(f[k-num]!=mx) ans=min(ans,f[k-num]+pnum);
    for(int i=hed[x];i!=0;i=nex[i])
    if(lb[i]!=fa && !pa[lb[i]])
    dfs1(lb[i],x,num+val[i],pnum+1);
}
void dfs2(int x,int fa,int num,int pnum){
    if(num>k) return ;
    update(num);f[num]=min(f[num],pnum);
    for(int i=hed[x];i!=0;i=nex[i])
    if(lb[i]!=fa && !pa[lb[i]])
    dfs2(lb[i],x,num+val[i],pnum+1);
}
void solve(int x){
    T++;f[0]=0;tcl[0]=T;
    sum=get_siz(x,x);
    int r0=getu(x,x);
    for(int i=hed[u];i!=0;i=nex[i])
    if(!pa[lb[i]]) {
        dfs1(lb[i],u,val[i],1);
        dfs2(lb[i],u,val[i],1);
    }
    pa[u]=1;
    for(int i=hed[u];i!=0;i=nex[i])
    if(!pa[lb[i]]) solve(lb[i]);
}
int main(){
    int ai,bi,ci;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=k;i++) f[i]=mx;
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&ai,&bi,&ci);ai++;bi++;
        add(ai,bi,ci);add(bi,ai,ci);    
    }
    solve(1);
    if(ans==mx) printf("-1");
    else printf("%d",ans);
    return 0;
}