【模板】点分治

洛谷 3806 

#include<cstdio>
#include<algorithm>
#define N 10010
#define rg register
using namespace std;
int n,m,tot,cnt,root,last[N],siz[N],mxsiz[N],dep[N],q[N];
bool have[10000010],ans[110],v[N];
struct edge{
    int to,pre,dis;
}e[N<<2];
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;
}
void getroot(int x,int fa){
    siz[x]=1; 
    for(rg int i=last[x],to;i;i=e[i].pre)if(!v[to=e[i].to]&&to!=fa){
        getroot(to,x); 
        siz[x]+=siz[to]; mxsiz[x]=max(mxsiz[x],siz[to]);
    }
    mxsiz[x]=max(mxsiz[x],cnt-siz[x]);
    if(!root||mxsiz[x]<mxsiz[root]) root=x;
}
void getdep(int x,int fa,int d){
    dep[tot++]=d;
    for(rg int i=last[x],to;i;i=e[i].pre)
    if(!v[to=e[i].to]&&to!=fa) getdep(to,x,d+e[i].dis);
}
void dfs(int x){
    v[x]=1; int l=tot=1; have[0]=1;
    for(rg int i=last[x],to;i;i=e[i].pre)if(!v[to=e[i].to]){
        l=tot; getdep(to,x,e[i].dis);
        for(rg int j=1;j<=m;j++){
            if(ans[j]) continue;
            for(rg int k=l;k<tot;k++)if(q[j]>=dep[k]&&have[q[j]-dep[k]]){
                ans[j]=1; break;
            }
        }
        for(rg int j=l;j<tot;j++) have[dep[j]]=1;
    }
    for(rg int i=1;i<tot;i++) have[dep[i]]=0;
    for(rg int i=last[x],to;i;i=e[i].pre)if(!v[to=e[i].to]){
        cnt=siz[to]; root=0; getroot(to,x); dfs(root);
    }
}
int main(){
    n=read(); m=read();
    for(rg int i=1;i<n;i++){
        int u=read(),v=read(),w=read();
        e[++tot]=(edge){v,last[u],w}; last[u]=tot;
        e[++tot]=(edge){u,last[v],w}; last[v]=tot;
    }
    for(rg int i=1;i<=m;i++) q[i]=read();
    root=0; cnt=n; getroot(1,0); dfs(root);
    for(rg int i=1;i<=m;i++) puts(ans[i]?"AYE":"NAY");
    return 0;
}