IOI2011 Race

Description

给一棵有N（1 <= N <= 200000）个结点的树，每条边有权，求一条路径，权值和等于K（1 <= K <= 1000000）且边的数量最小。

Input

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

Output

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

Sample Input

4 3

0 1 1

1 2 2

1 3 4

Sample Output

2

#include<bits/stdc++.h>
using namespace std;
const int Maxn=200005;
struct Edge{
	int cnt,h[Maxn],w[Maxn*2],to[Maxn*2],next[Maxn*2];
	inline void add(int x,int y,int z){
		next[++cnt]=h[x];to[cnt]=y;w[cnt]=z;h[x]=cnt;
	}
}e;
struct Node{
	int v,num;
	bool operator <(const Node&A) const {
		return v<A.v||v==A.v&&num<A.num;
	}
}q[Maxn];set<Node>s;
#define to e.to[p]
int cnt,ans=1<<30;
int tmpsiz[Maxn];
bool vst[Maxn];
inline void stat(int x,int fa,int num,int dist){
	q[++cnt]=(Node){dist,num};
	for(int p=e.h[x];p;p=e.next[p])
		if(!vst[to]&&(to^fa))stat(to,x,num+1,dist+e.w[p]);
}
inline void work(int x,int goal){
	s.clear();
	for(int p=e.h[x];p;p=e.next[p])if(!vst[to]){
		cnt=0;stat(to,x,1,e.w[p]);
		for(int i=1;i<=cnt;++i){
			Node nd=*s.lower_bound((Node){goal-q[i].v,0});
			if(nd.v+q[i].v==goal&&s.count(nd))ans=min(ans,nd.num+q[i].num);
		}
		for(int i=1;i<=cnt;++i)s.insert(q[i]);
	}
	Node nd=*s.lower_bound((Node){goal,0});
	if(nd.v==goal&&s.count(nd))ans=min(ans,nd.num);
}
inline void getroot(int x,int fa,int &mn,int &root,int totsiz){
	tmpsiz[x]=1;int maxsiz=0;
	for(int p=e.h[x];p;p=e.next[p])if(!vst[to]&&(to^fa)){
		getroot(to,x,mn,root,totsiz);
		tmpsiz[x]+=tmpsiz[to];
		maxsiz=max(maxsiz,tmpsiz[to]);
	}
	maxsiz=max(maxsiz,totsiz-tmpsiz[x]);
	if(maxsiz<mn)mn=maxsiz,root=x;
}
inline void Divide(int x,int goal,int totsiz){
	int mn=1<<30,root=x;
	getroot(x,0,mn,root,totsiz);
	work(root,goal);
	vst[root]=1;
	for(int p=e.h[root];p;p=e.next[p])
		if(!vst[to])Divide(to,goal,tmpsiz[to]);
}
int main(){
	int n,k;scanf("%d%d",&n,&k);
	for(int i=1;i<n;++i){
		int x,y,z;scanf("%d%d%d",&x,&y,&z);
		e.add(x+1,y+1,z),e.add(y+1,x+1,z);
	}
	Divide(1,k,n);
	printf("%d\n",ans==1<<30?-1:ans);
	return 0;
}