WA on #9 求调

#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<cstring>
#include<algorithm>
#include<cassert>
#include<stack>
#include<queue>
#include<vector>
#include<map>
#include<cstdlib>
#include<set>
using namespace std;
#define ll long long
#define ull unsigned long long
int read()
{
	int x=0,f=1;
	char ch=getchar();
	while(ch<'0' || ch>'9')
	{
		if(ch=='-')
			f=-1;
		ch=getchar();
	}
	while(ch>='0' && ch<='9')
	{
		x=x*10+(ch-'0');
		ch=getchar();
	}
	return x*f;
}
const int MAXN=1e6+10;
int n,m;
int q[MAXN];
int head[MAXN],tt=0;
struct edge
{
	int to,nxt,dis;
}e[MAXN<<1];
void add(int x,int y,int z)
{
	e[++tt].nxt=head[x];
	head[x]=tt;
	e[tt].to=y;
	e[tt].dis=z;
}
int root;//表示树的重心 
int tot;
int siz[MAXN],maxn[MAXN];
bool vis[MAXN];
void dfs(int u,int fa)//求树的重心 
{
	siz[u]=1;
	maxn[u]=0;
	for(int i=head[u];i;i=e[i].nxt)
	{
		int v=e[i].to,w=e[i].dis;
		if(v==fa || vis[v])
			continue;
		dfs(v,u);
		siz[u]+=siz[v];
		maxn[u]=max(maxn[u],siz[v]);
	}
	maxn[u]=max(maxn[u],tot-siz[u]);
	if(maxn[u]<maxn[root])
		root=u;
}
int dis[MAXN];
int seq[MAXN],cnt=0;
void dfs1(int u,int fa)//处理dis数组，即每个点的树的重心的距离 
{
	seq[++cnt]=dis[u];
	for(int i=head[u];i;i=e[i].nxt)
	{
		int v=e[i].to,w=e[i].dis;
		if(v==fa || vis[v])
			continue;
		dis[v]=dis[u]+w;
		dfs1(v,u);
	}
}
bool tmp[MAXN],ans[MAXN];
int qu[MAXN];
void calc(int u)
{
	int c=0;
	for(int i=head[u];i;i=e[i].nxt)
	{
		int v=e[i].to,w=e[i].dis;
		if(vis[v])
			continue;
		cnt=0;
		dis[v]=w;
		dfs1(v,u);
		for(int j=cnt;j;j--)
		{
			for(int k=1;k<=m;k++)
			{
				if(q[k]>=seq[j])
					ans[k]|=tmp[q[k]-seq[j]];
			}
		}
		for(int j=cnt;j;j--)
			qu[++c]=seq[j],tmp[seq[j]]=1;
	}
	for(int i=1;i<=c;i++)
		tmp[qu[i]]=0;
}
void solve(int u)
{
	vis[u]=tmp[0]=1;
	calc(u);
	for(int i=head[u];i;i=e[i].nxt)
	{
		int v=e[i].to;
		if(vis[v])
			continue;
		maxn[0]=n,tot=siz[v],root=0;
		dfs(v,u),solve(root);
	}
}
int main()
{
	n=read(),m=read();
	for(int i=1;i<n;i++)
	{
		int x=read(),y=read(),z=read();
		add(x,y,z),add(y,x,z);
	}
	for(int i=1;i<=m;i++)
		q[i]=read();
	maxn[root]=n,tot=n;
	dfs(1,0);
	solve(root);//从树的重心开始向下分治遍历
	for(int i=1;i<=m;i++)
	{
		if(ans[i])
			printf("AYE\n");
		else
			printf("NAY\n");
	}
	return 0;
}