[FROM WOJ]#3968 [IOI2011] Race

#3968 [IOI2011] Race

题面
给一棵树,每条边有权.求一条路径,权值和等于K,且边的数量最小.

输入
第一行 两个整数 n, k
第二…n行 每行三个整数 表示一条无向边的两端和权值 (注意点的编号从0开始)

输出
一个整数 表示最小边数量 如果不存在这样的路径 输出-1

样例输入
4 3
0 1 1
1 2 2
1 3 4

样例输出
2

数据规模
$n\le 2e5，k\le 1e6$

SOL
点分治模板题。
对于每次分治，先求出每个点到分治中心的距离（同时可以维护边数，即Dep），然后对距离排序，用双指针统计一下答案。
需要注意的是，需要记录每个点的对应子树——如果在分治中心的同一子树显然是不成立的。

代码：

#include<bits/stdc++.h>
using namespace std;
#define re register
namespace Base{
	template<typename T>inline void chmin(T &a,T b){return a<b?a:b;}
	template<typename T>inline void chmax(T &a,T b){return a>b?a:b;}
}using namespace Base;
inline char nc(){
    static char buf[1000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000,stdin),p1==p2)?EOF:*p1++;
}
#define getchar nc
inline int rd(){
	int re data=0,w=1;static char ch=0;ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))data=(data<<1)+(data<<3)+(ch^48),ch=getchar();
	return data*w;
}
inline void write(int x){if(x<0)x=-x,putchar('-');if(x>9)write(x/10);putchar(x%10+'0');}
const int N=2e5+5;
int n,cnt,first[N],siz[N],ans=1e9+7,root,tot,d[N],K,num,f[N];
bool mark,flag[N];
struct edge{int v,nxt,w;}e[N<<1];
struct pt{int dep,l,id;}p[N];
inline void add(int u,int v,int w){e[++cnt]=(edge){v,first[u],w};first[u]=cnt;}
inline void getG(int u,int fa){
	siz[u]=1,f[u]=0;
	for(int re i=first[u];i;i=e[i].nxt){
		int re v=e[i].v;
		if(v==fa||flag[v])continue;
		getG(v,u),siz[u]+=siz[v],f[u]=max(f[u],siz[v]);
	}f[u]=max(f[u],tot-siz[u]);
	if(f[u]<f[root])root=u;
}
inline void getdep(int top,int u,int dep,int fa){
	p[++num].dep=d[u],p[num].l=dep;
	if(fa==top||!fa)p[num].id=u;else p[num].id=p[num-1].id;
	for(int re i=first[u];i;i=e[i].nxt){
		int re v=e[i].v;
		if(v==fa||flag[v])continue;
		d[v]=d[u]+e[i].w,getdep(top,v,dep+1,u);
	}
}
inline bool cmp(pt a,pt b){return a.dep<b.dep;}
inline void calc(int u){
	num=0,d[u]=0;bool re ret=0;
	getdep(u,u,0,0),sort(p+1,p+num+1,cmp);
	for(int re l=1,r=num;l<r;){
		if(p[l].dep+p[r].dep==K){
			int re p1=l,p2=r;
			while(p[p1].dep+p[r].dep==K)p1++;p1--;
			while(p[p2].dep+p[l].dep==K)p2--;p2++;
			for(int re j=l;j<=p1;j++)
				for(int re k=p2;k<=r;k++)//特殊情况：两个点同属一颗子树->不合法 
					if(p[j].id!=p[k].id)ans=min(ans,p[j].l+p[k].l),ret=1;
			l=p1+1,r=p2-1;
		}else if(p[l].dep+p[r].dep<K)l++;else if(p[l].dep+p[r].dep>K)r--;
	}mark|=ret;
}
inline void sol(int u){
	calc(u),flag[u]=1;//标记分支中心，表示删除这个点及其连边 
	for(int re i=first[u];i;i=e[i].nxt){
		int re v=e[i].v;
		if(flag[v])continue;
		f[0]=tot=siz[v],getG(v,root=0),sol(root);
	}
}
signed main(){
	n=rd(),K=rd();
	for(int re i=1;i<n;i++){
		int re u=rd()+1,v=rd()+1,w=rd();
		add(u,v,w),add(v,u,w);
	}f[0]=tot=n,getG(1,root=0),sol(root),write(mark?ans:-1),exit(0);
}