淀粉质WA on#7 求条

#include<bits/stdc++.h>
// #define int long long
using namespace std;
const int maxn=3e5+5;
vector<pair<int,int> > vec[maxn];
int n,k;
map<long long,int> t1,t2;
bool flag[maxn];
int ans=INT_MAX;
int getsz(int u,int fa){
	int ans=1;
	for(auto p:vec[u]){
		int v=p.first;
		if(v==fa||flag[v])continue;
		ans+=getsz(v,u);
	}
	return ans;
}
int minw,w;
int getwc(int u,int fa,int num){
	int ans=1;int maxw=0;
	for(auto pp:vec[u]){
		int v=pp.first;
		if(v==fa||flag[v])continue;
		int p=getwc(v,u,num);
		ans+=p;
		maxw=max(maxw,p);
	}
	maxw=max(maxw,num-ans);
	if(maxw<minw){
		minw=maxw;
		w=u;
	}
	return ans;
}
void judge(int u,int fa,int dep,int dis){
	if(t1.find(dis)==t1.end()||t1[dis]>dep)t1[dis]=dep;
	for(auto p:vec[u]){
		int v=p.first,w=p.second;
		if(v==fa||flag[v])continue;
		judge(v,u,dep+1,dis+w);
	}
}
void calc(int u){
	t2.clear();
	t2[0]=0;
	int x=getsz(u,-1)-1;
	w=u,minw=x;
	getwc(u,-1,x+1);
	flag[w]=1;
	for(auto pp:vec[w]){
		int v=pp.first,ww=pp.second;
		if(flag[v])continue;
		judge(v,w,1,ww);
		for(auto p:t1){
			if(t2.find(k-p.first)!=t2.end()){
				ans=min(ans,t2[k-p.first]+p.second);
			}
		}
		for(auto p:t1)t2[p.first]+=p.second;
		t1.clear();
	}
	for(auto p:vec[w]){
		int v=p.first;
		if(flag[v])continue;
		calc(v);
	}
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	cin>>n>>k;
	for(int i=1;i<n;i++){
		int u,v,w;
		cin>>u>>v>>w;
		vec[u].emplace_back(v,w);
		vec[v].emplace_back(u,w);
	}
	calc(0);
	if(ans!=INT_MAX)cout<<ans<<endl;
	else cout<<-1<<endl;
	return 0;
}