点分治模板

分治思想：一直找子树的重心，使每次遍历的子树深度尽量的小。

#include<bits/stdc++.h>
using namespace std;
const int N=1e7+10;
const int inf=0x3f3f3f3f;
typedef long long ll;
typedef pair<int,int> pii;
typedef unsigned long long ull;
//const ll P=2281701377;
const ll P=998244353;
//#define int ll

int n,m;
vector<int> v[100010],l[100010];
int ans[1010],query[1010];
bool del[100010],judge[N];
int tempd[100010],dis[100010],cnt;
int sum,maxsiz[100010],siz[100010],root;
void get_root(int x,int f){
	siz[x]=1;maxsiz[x]=0;
	for(auto j:v[x]){
		if(j==f||del[j]) continue;
		get_root(j,x);
		siz[x]+=siz[j];
		maxsiz[x]=max(maxsiz[x],siz[j]);
	}
	maxsiz[x]=max(maxsiz[x],sum-siz[x]);
	if(maxsiz[x]<maxsiz[root])
		root=x;//sum-siz【x】表示如果删除当前节点划分出来的父亲节点那块的大小
                //maxsiz【x】表示以当前节点为根的多个子树里最大的子树大小
}
void get_dis(int x,int f){
	tempd[++cnt]=dis[x];
	for(int i=0;i<v[x].size();i++){
		int j=v[x][i],w=l[x][i];
		if(j==f||del[j]) continue;
		dis[j]=dis[x]+w;
		get_dis(j,x);
	}
}
queue<int> q;
void calc(int x){
	for(int i=0;i<v[x].size();i++){
		int j=v[x][i],w=l[x][i];
		if(del[j]) continue;//删掉的点不要算
		cnt=0;
		dis[j]=w;
		get_dis(j,x);//更新子树的每个点到当前根节点的距离
		for(int k=1;k<=cnt;k++)
			for(int t=1;t<=m;t++)
				if(query[t]>=tempd[k])
					ans[t]|=judge[query[t]-tempd[k]];
		for(int k=1;k<=cnt;k++){
			if(tempd[k]>=N-5) continue;//为了防止越界，一定要加这个判断
			q.push(tempd[k]);
			judge[tempd[k]]=1;
		}
	}
	while(q.size()){//用queue来临时存储和用完后清空
		judge[q.front()]=0;
		q.pop();
	}
}
void divide(int x){
	del[x]=judge[0]=1;//用过x就把x  del删掉，根到根距离为0，所以0肯定可以到
	calc(x);//每次枚举一个子树的时候，都把这颗子树的贡献算上（就是一定经过当前根的距离）
	for(auto j:v[x]){
		if(del[j]) continue;//如果删掉了就不再对这颗树进行分治
		sum=maxsiz[root=0]=siz[j];//每次分治的时候都要重新找重心，需要重置root和最大值
		get_root(j,0);//找重心
		get_root(root,0);//一定要以重心重新遍历一遍来更新正确的siz值
		divide(root);
	}
}
void solve(){
	cin>>n>>m;
	for(int i=1;i<n;i++){
		int a,b,c;
		cin>>a>>b>>c;
		v[a].push_back(b);
		v[b].push_back(a);
		l[a].push_back(c);
		l[b].push_back(c);
	}
	for(int i=1;i<=m;i++)
		cin>>query[i];
	sum=maxsiz[root=0]=n;
	get_root(1,0);//找重心
	get_root(root,0);//找到重心之后一定要以重心再进行一遍，不然子树的siz并不是以重心为根的子树siz
	divide(root);//以重心进去分治
	for(int i=1;i<=m;i++){
		if(ans[i]){
			cout<<"AYE"<<endl;
		}
		else{
			cout<<"NAY"<<endl;
		}
	}
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    int t=1;
    //cin>>t;
    while(t--){
        solve();
    }

}