正题

评测记录:https://www.luogu.org/recordnew/lists?uid=52918&pid=P3806

---

题目大意

一颗树，k个询问，求是否存在长度为$l_i$的路径。

---

解题思路

开个桶就好了，点分治不解释。

---

$code$

#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 100010
#define inf 10000000
using namespace std;
struct node{int to,next,w;
}a[N*2];
int n,m,que[1010],ok[inf],tot,sum,num,q[N];
int root,siz[N],f[N],ls[N],st[N],dis[N]; 
int lon[inf];
bool v[N];
void addl(int x,int y,int w)
{a[++tot].to=y;a[tot].next=ls[x];a[tot].w=w;ls[x]=tot;
}
void groot(int x,int fa)
{siz[x]=1;f[x]=0;for(int i=ls[x];i;i=a[i].next){int y=a[i].to;if(y==fa||v[y]) continue;groot(y,x);siz[x]+=siz[y];f[x]=max(f[x],siz[y]);}f[x]=max(f[x],sum-siz[x]);if(f[x]<f[root]) root=x;
}
void get_dis(int x,int fa)
{st[++num]=dis[x];for(int i=ls[x];i;i=a[i].next){int y=a[i].to;if(y==fa||v[y]) continue;dis[y]=dis[x]+a[i].w;get_dis(y,x);}
}
void calc(int x)
{int p=0;for(int i=ls[x];i;i=a[i].next){int y=a[i].to;if(v[y]) continue;num=0;dis[y]=a[i].w;get_dis(y,x);for(int j=num;j;j--)for(int k=1;k<=m;k++)ok[k]|=lon[que[k]-st[j]];for(int j=num;j;j--)q[++p]=st[j],lon[st[j]]=1;}for(int i=1;i<=p;i++)lon[q[i]]=0;
}
void solve(int x)
{v[x]=lon[0]=1;calc(x);for(int i=ls[x];i;i=a[i].next){int y=a[i].to;if(v[y]) continue;sum=siz[y];f[root=0]=n;groot(y,0);solve(y);}
}
int main()
{scanf("%d%d",&n,&m);for(int i=1;i<n;i++){int x,y,w;scanf("%d%d%d",&x,&y,&w);addl(x,y,w);addl(y,x,w);}for(int i=1;i<=m;i++)scanf("%d",&que[i]);f[0]=n;sum=n;groot(1,0);solve(root);for(int i=1;i<=m;i++)if(ok[i]) printf("AYE\n");else printf("NAY\n");
}