#include<bits/stdc++.h>
using namespace std;
int size[10001],mxsize[10001],k[101];
bool ans[101],del[10001],len[10000001];//del用来标记删除。len是桶,用来存储距离。
vector<int>data;
queue<int>que,nodes;
struct edge{
int end,val;
};
vector<edge>tree[10001];
void dfs(int root,int fa)//求子树大小
{
nodes.push(root);
size[root]=1;
mxsize[root]=0;
for(int i=0;i<tree[root].size();i++)
if(tree[root][i].end!=fa&&!del[tree[root][i].end])
{
dfs(tree[root][i].end,root);
size[root]+=size[tree[root][i].end];
mxsize[root]=max(mxsize[root],size[tree[root][i].end]);
}
}
int getcenter(int root)//求重心
{
int x,pos,mx,smx;
dfs(root,0);
mx=-1;
while(!nodes.empty())
{
x=nodes.front();
nodes.pop();
smx=max(size[root]-size[x],mxsize[x]);
if(smx>mx)
{
mx=smx;
pos=x;
}
}
return pos;
}
void getdis(int root,int fa,int dis)//获取根结点到其他结点的距离
{
data.push_back(dis);
for(int i=0;i<tree[root].size();i++)
if(tree[root][i].end!=fa&&!del[tree[root][i].end])
getdis(tree[root][i].end,root,dis+tree[root][i].val);
}
void divide(int root,int m)//点分治
{
int x;
root=getcenter(root);
for(int i=0;i<tree[root].size();i++)
if(!del[tree[root][i].end])
{
getdis(tree[root][i].end,root,tree[root][i].val);
for(int j=0;j<data.size();j++)
for(int l=1;l<=m;l++)
if(data[j]<=k[l])
if(len[k[l]-data[j]])
ans[l]=true;
for(int j=0;j<data.size();j++)
if(data[j]<=10000000)
{
len[data[j]]=true;
que.push(data[j]);
}
data.clear();
}
while(!que.empty())//清空桶
{
x=que.front();
que.pop();
len[x]=false;
}
del[root]=true;
for(int i=0;i<tree[root].size();i++)
if(!del[tree[root][i].end])
divide(tree[root][i].end,m);
}
int main()
{
int n,m,u,v,w;
edge e;
cin>>n>>m;
for(int i=1;i<n;i++)
{
cin>>u>>v>>w;
e.val=w;
e.end=v;
tree[u].push_back(e);
e.end=u;
tree[v].push_back(e);
}
len[0]=true;
for(int i=1;i<=m;i++)//离线
cin>>k[i];
divide(1,m);
for(int i=1;i<=m;i++)
if(ans[i]||k[i]==0)//对询问为0的情况进行特判
cout<<"AYE"<<endl;
else
cout<<"NAY"<<endl;
return 0;
}
#7#8#9#10TLE了,但我没有看出来我的代码哪里有问题。时间复杂度为O(nm log n)。