解题思路
点分治练习。
注意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;
}