树分治--点分治模板

存个板子，内容以后补

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e4+40,inf=1e9;
int n,m,rt;
int head[maxn<<1],ecnt=-1,siz[maxn],dist[maxn],dd[maxn];
int maxx[maxn]; 
int q[maxn],sum,cnt;
bool tf[10001000],ret[maxn],vis[maxn];
queue<int> tag;

struct mint
{
	int nxt,v,w;	
}e[maxn<<1];
inline void addline(int u,int v,int w)
{
	e[++ecnt].nxt=head[u];
	e[ecnt].v=v;
	e[ecnt].w=w;
	head[u]=ecnt;	
}

void calcsiz(int x,int fa)
{
	siz[x]=1;
	maxx[x]=0;
	for(int i=head[x];~i;i=e[i].nxt)
	{
		int v=e[i].v;
		if(v==fa) continue;
		if(vis[v]) continue;
		calcsiz(v,x);	
		maxx[x]=max(maxx[x],siz[v]);
		siz[x]+=siz[v];
	}
	maxx[x]=max(maxx[x],sum-siz[x]);
	if(maxx[x]<maxx[rt]) rt=x;
}	

void calcdist(int x,int fa)
{
	dd[++cnt]=dist[x];
	for(int i=head[x];~i;i=e[i].nxt)
	{
		int v=e[i].v;
		if(v==fa) continue;
		if(vis[v]) continue;
		dist[v]=dist[x]+e[i].w;
		calcdist(v,x);
	}
}

void dfz(int x,int fa)
{
	tf[0]=true;
	tag.push(0);
	vis[x]=true;
	for(int i=head[x];~i;i=e[i].nxt)
	{
		int v=e[i].v;
		if(v==fa) continue;
		if(vis[v]) continue;
		dist[v]=e[i].w;
		calcdist(v,x);
		for(int k=1;k<=cnt;++k)
			for(int i=1;i<=m;++i)
				if(q[i]>=dd[k]) ret[i] |= tf[q[i]-dd[k]];
		for(int k=1;k<=cnt;++k)
			if(dd[k]<10000010) tag.push(dd[k]),tf[dd[k]]=true;
		cnt=0;
	}
	while(!tag.empty()) tf[tag.front()]=false,tag.pop();
	for(int i=head[x];~i;i=e[i].nxt)
	{
		int v=e[i].v;
		if(v==fa) continue;
		if(vis[v]) continue;
		sum=siz[v];
		rt=0;
		maxx[rt]=inf;
		calcsiz(v,x);
		calcsiz(rt,-1);
		dfz(rt,x);		
	}
}

int main()
{
	memset(head,-1,sizeof(head)); 
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n-1;++i) 
	{
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		addline(u,v,w);
		addline(v,u,w);	
	}
	for(int i=1;i<=m;++i) scanf("%d",&q[i]);
	rt=0;
	maxx[rt]=inf;
	sum=n;
	calcsiz(1,-1);
	calcsiz(rt,-1);
	dfz(rt,-1);
	for(int i=1;i<=m;++i)
	{
		if(ret[i]) printf("AYE\n");
		else printf("NAY\n");	
	}
	return 0;
}