[模板]点分治

算法流程：找到树的重心，删去这个点，递归子树。
题号luogu3806

#include <iostream>
#include <ctime>
#include <algorithm>
#include <windows.h>
#include <cstdio>
#include <cstring>
using namespace std;
const int N=10005;
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
}
int n,m,K,siz[N],head[N],ecnt,size,rt,cnt,a[150],f[N];
struct Edge{int to,nxt,val;}e[N<<1];
struct Node{int from,val; bool operator < (const Node &rhs) const {return val<rhs.val;};};
bool vis[N],ans[106];
Node dis[10005];
void add(int bg,int ed,int val){(++ecnt)[e].to=ed;ecnt[e].val=val;ecnt[e].nxt=bg[head];bg[head]=ecnt;}
int binary(int x) {
    int l=1,r=cnt,ans=0;
    while(l<=r) {
        int mid=l+r>>1;
        if(dis[mid].val<x) l=mid+1;
        else r=mid-1,ans=mid;
    }
    return ans;
}
void barycenter(int x,int fa) {
    f[x]=0,siz[x]=1;
    for(int i=head[x];i;i=e[i].nxt) {
        int v=e[i].to;
        if(vis[v]||fa==v) continue;
        barycenter(v,x);
        f[x]=max(f[x],siz[v]);
        siz[x]+=siz[v];
    }
    f[x]=max(f[x],size-siz[x]);
    if(f[x]<f[rt]) rt=x;
}
void dfs(int x,int fa,int from,int val) {
    dis[++cnt]={from,val};
    for(int i=head[x];i;i=e[i].nxt) {
        int v=e[i].to;
        if(vis[v]||fa==v) continue;
        dfs(v,x,from,val+e[i].val);
    }
}
void work(int x) {
    cnt=0;
    for(int i=x[head];i;i=i[e].nxt ) {
        int v=i[e].to;
        if(v[vis]) continue;
        dfs(v,x,v,e[i].val);
    }
    (++cnt)[dis]={0,0};
    sort(dis+1,dis+1+cnt);
    for(int i=1;i<=m;i++) {
        if(i[ans]) continue;
        int l=1;
        while(l<cnt&&a[i]>(dis[l].val+dis[cnt].val)) l++;
        while(l<cnt&&!ans[i]) {
            if(a[i]-dis[l].val<dis[l].val) break;
            int pos=binary(a[i]-dis[l].val);
            while(dis[pos].val+dis[l].val==a[i]&&l<=cnt&&dis[pos].from==dis[l].from) pos++;
            if(dis[pos].val+dis[l].val==a[i]) ans[i]=1;
            l++;
        }
    }
}
void solve(int x) {
    vis[x]=1;
    work(x);
    for(int i=head[x];i;i=e[i].nxt) {
        int v=e[i].to;rt=0;
        if(v[vis]) continue;
        siz[rt]=size=siz[v];
        barycenter(v,0);
        solve(rt);
    }
}
int main() {
    int st=clock();
    n=read();m=read();
    for(int i=1,x,y,z;i<n;i++)  x=read(),y=read(),z=read(),add(x,y,z),add(y,x,z);
    for(int i=1;i<=m;i++) a[i]=read();
    f[rt]=size=n;
    barycenter(1,0);
    solve(rt);
    for(int i=1;i<=m;i++)puts(ans[i]?"AYE":"NAY");
    int ed=clock();
    Sleep(200);
    cout<<ed-st;
    return 0;
}