luoguP3806 【模板】点分治1

#include<bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin) 
#define maxn 10004  
#define inf 10000003
using namespace std;
int edges,n,Q,sn,root,tl; 
bool is[inf]; 
int hd[maxn],to[maxn<<1],nex[maxn<<1],val[maxn<<1];  
int answer[maxn], que[200], vis[maxn], f[maxn], siz[maxn], dep[maxn], mine[inf], dis1[maxn];        
inline void add(int u,int v,int c)
{
	nex[++edges]=hd[u],hd[u]=edges,to[edges]=v,val[edges]=c; 
} 
void Getroot(int u,int fa)
{
	f[u]=0, siz[u]=1; 
	for(int i=hd[u];i;i=nex[i])
	{
		int v=to[i]; 
		if(vis[v]||v==fa) continue; 
		Getroot(v,u), siz[u]+=siz[v]; 
		f[u]=max(f[u], siz[v]); 
	}
	f[u]=max(f[u], sn-siz[u]); 
	if(f[u]<f[root]) root=u;     
}
inline void getdis(int u,int fa,int d)
{      
	dis1[++tl] = d;  
	for(int i=hd[u];i;i=nex[i])
	{ 
		int v=to[i]; 
		if(v==fa||vis[v]) continue; 
		getdis(v, u, d + val[i]);   
	}
}
inline void calc(int u)
{
	tl=0; 
	mine[0]=1;        
	for(int i=hd[u];i;i=nex[i])
	{
		int v=to[i];
		if(vis[v]) continue; 
		int pdl=tl; 
		getdis(v, u, val[i]); 
		for(int j=pdl+1;j<=tl;++j) 
			for(int o=1;o<=Q;++o)
			{
				if(que[o]>=inf||que[o] < dis1[j]) continue;  
				is[que[o]]|=mine[que[o]-dis1[j]];     
			}   
		for(int j=pdl+1;j<=tl;++j) mine[dis1[j]]=1; 
	} 
    for(int i=1;i<=tl;++i) mine[dis1[i]]=0;      
}
void solve(int u)
{
	int i,v;
	vis[u]=1;    
	calc(u);     
	for(i=hd[u];i;i=nex[i])
	{
		v=to[i]; 
		if(vis[v]) continue;   
		root=0,sn=siz[v],Getroot(v, u); 
		solve(root);      
	}
}
int main()
{
	int i,j; 
	// setIO("input"); 
	scanf("%d%d",&n,&Q);
	for(i=1;i<n;++i)
	{
		int u,v,c; 
		scanf("%d%d%d",&u,&v,&c); 
		add(u,v,c), add(v,u,c);   
	}
	for(i=1;i<=Q;++i) scanf("%d",&que[i]);  
	sn=n,f[0]=maxn,Getroot(1,0),solve(root);  
	for(i=1;i<=Q;++i) if(is[que[i]]) puts("AYE"); else puts("NAY");   
	return 0; 
}