第一次写树分治.
刚开始被属于同一分支的点通过根组成的链坑了一发.
然后又被数组下标从零开始坑了一发.
最后数组大小多写了个0又MLE了一发.
感觉有点难看将就一下吧.
#include <queue>
#include <cstdio>
using namespace std;
const int N=200050,M=1000050;
int n,K;
int gfa,rt,tot,ans=1<<30;
int v[M],e[M];
int d[N],dis[N],dep[N],mxs[N],siz[N];
bool vis[N];
struct edge {
int u,v,w;
edge *nxt;
edge (int u,int v,int w,edge *pre): u(u),v(v),w(w){nxt=pre;}
};
edge *he[N*2];
queue <int> q;
void grt(int u,int fa) {
mxs[u]=0; siz[u]=1;
for (edge *i=he[u];i;i=i->nxt) if (i->v!=fa && !vis[i->v]) {
grt(i->v,u);
siz[u]+=siz[i->v];
mxs[u]=max(mxs[u],siz[i->v]);
}
mxs[u]=max(mxs[u],siz[gfa]-siz[u]);
if (mxs[u]<mxs[rt]) rt=u;
}
void DFS(int u,int fa) {
d[++tot]=u;
for (edge *i=he[u];i;i=i->nxt) if (i->v!=fa && !vis[i->v]) {
dis[i->v]=dis[u]+i->w; dep[i->v]=dep[u]+1; DFS(i->v,u);
}
}
void dfz(int u) {
dis[u]=0; dep[u]=0;
for (edge *i=he[u];i;i=i->nxt) if (!vis[i->v]) {
tot=0; v[0]=u; gfa=u; dis[i->v]=dis[u]+i->w; dep[i->v]=dep[u]+1; DFS(i->v,0);
for (int j=1;j<=tot;j++) if (K-dis[d[j]]>=0)
if (v[K-dis[d[j]]]==gfa) ans=min(ans,e[K-dis[d[j]]]+dep[d[j]]);
for (int j=1;j<=tot;j++) if (dis[d[j]]<K)
if (v[dis[d[j]]]==gfa) e[dis[d[j]]]=min(dep[d[j]],e[dis[d[j]]]);
else v[dis[d[j]]]=gfa, e[dis[d[j]]]=dep[d[j]];
}
for (edge *i=he[u];i;i=i->nxt) if (!vis[i->v]) {
gfa=u; rt=0; grt(i->v,0); vis[rt]=1; if (siz[rt]>1) dfz(rt);
}
}
int main() {
int i,a,b,c;
scanf("%d%d",&n,&K); siz[0]=mxs[0]=n;
for (i=1;i<n;i++) {
scanf("%d %d %d",&a,&b,&c); a++; b++;
he[a]=new edge(a,b,c,he[a]);
he[b]=new edge(b,a,c,he[b]);
}
gfa=0; rt=0; grt(1,0); vis[rt]=1; dfz(rt);
if (ans==1<<30) ans=-1;
printf("%d",ans);
}