题目链接：点这里

题目描述：给你一棵n个点的带权有根树，有p个询问，每次询问树中是否存在一条长度为Len的路径，如果是，输出Yes否输出No.

数据范围：\(n\le1e5\,,p\le100\,,长度\le1e5\)

Solution:

点分治裸题，没什么好讲的。不过注意当询问0时，答案应该是Yes。

Code:

#pragma GCC optimize(3)
#include<bits/stdc++.h>
#define ll long long
#define inf 2147483647
using namespace std;
const int N=1e4+1;
int rt,rtu,cnt,szt,head[N],sz[N],mx[N],rtt[N];
int n,m,k,ans,l,r,tot,dis[N],vis[N];
struct Edge{int nxt,to,val;}edge[N<<1];
void ins(int x,int y,int z){edge[++cnt].nxt=head[x];edge[cnt].to=y;edge[cnt].val=z;head[x]=cnt;
}
void getrt(int x,int f){sz[x]=1,mx[x]=0;for(int i=head[x];i;i=edge[i].nxt){int y=edge[i].to;if(y==f||vis[y]) continue;getrt(y,x);sz[x]+=sz[y];mx[x]=max(mx[x],sz[y]);}mx[x]=max(mx[x],szt-sz[x]);if(mx[x]<mx[rt]) rt=x;
}
void getdis(int x,int f,int d){dis[++tot]=d;for(int i=head[x];i;i=edge[i].nxt){int y=edge[i].to;if(y==f||vis[y]) continue;getdis(y,x,d+edge[i].val);}
}
int find(int x,int l,int r){while(l<r){int mid=(l+r)>>1;if(dis[mid]+x<k) l=mid+1;else r=mid;}return l;
}
int calc(int x,int d){tot=0;getdis(x,0,d);sort(dis+1,dis+tot+1);int num=0,l=1,r=tot;while(l<r){if(dis[l]+dis[r]>k) --r;else if(dis[l]+dis[r]<k) ++l;else{if(dis[l]==dis[r]){num+=(r-l+1)*(r-l)/2;break;}int u1=l,u2=r;while(dis[u1]==dis[l]) ++u1;while(dis[u2]==dis[r]) --u2;num+=(u1-l)*(r-u2);l=u1,r=u2;}}return num;
}
void divide(int x){ans+=calc(x,0);vis[x]=1;for(int i=head[x];i;i=edge[i].nxt){int y=edge[i].to;if(!vis[y]){ans-=calc(y,edge[i].val);divide(rtt[++rtu]);}}
}
void getrtt(int x){vis[x]=1;for(int i=head[x];i;i=edge[i].nxt){int y=edge[i].to;if(!vis[y]){rt=0,szt=sz[y],getrt(y,0);rtt[++rtu]=rt;getrtt(rt);}}
}
int solve(){memset(vis,0,sizeof(vis));ans=0;divide(rtt[rtu=0]);return ans;
}
int read(){int x=0,f=1;char ch=getchar();while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();}while(isdigit(ch)){x=x*10+ch-48;ch=getchar();}return x*f;
}
int main(){n=read(),m=read();for(int i=1;i<n;i++){int x=read(),y=read(),z=read();ins(x,y,z);ins(y,x,z);}szt=n;mx[rt]=inf;ans=0;getrt(1,0);rtt[0]=rt;getrtt(rt);for(int i=1;i<=m;i++){k=read();if(!k){puts("Yes");continue;}solve();if(ans) puts("Yes");else puts("No");}return 0;
}