洛谷P3806 【模板】点分治1

（本题）淀粉汁本质上就是树上应用分治思想。比如这个路径长度，可以分为过根的和存在于子树内的两种。

过根的直接算，子树内的可以递归处理，为了使递归效率更高，每次取子树的重心为根。

详细思路见代码注释：

#include<bits/stdc++.h>
#define maxn 100086
#define inf 0x3f3f3f3f
using namespace std;
struct onanii{
    int v,w,next;
}e[maxn*2];
int n,m,k,h[maxn],cnt;
int x,y,z,query[maxn],ans[maxn],judge[10000001];
int root,sz[maxn],vis[maxn],d[maxn],maxsize;
int q[maxn],rem[maxn];
void add(int u,int v,int w){
    e[++cnt].v=v;
    e[cnt].w=w;
    e[cnt].next=h[u];
    h[u]=cnt;
}
void getroot(int S,int u,int fa){//找重心
    sz[u]=1;
    int maxpart=0;
    for(int i=h[u];i;i=e[i].next){
        int v=e[i].v;
        if(v==fa||vis[v])continue;
        getroot(S,v,u);
        sz[u]+=sz[v];
        maxpart=max(maxpart,sz[v]);
    }
    maxpart=max(maxpart,S-sz[u]);
    if(maxpart<maxsize){
        maxsize=maxpart;
        root=u;
    }
}
void getdis(int u,int fa){
    if(d[u]<=10000000)//题目k的范围最大到1e7
    rem[++rem[0]]=d[u];//记录路径长度
    for(int i=h[u];i;i=e[i].next){
        int v=e[i].v;
        if(v==fa||vis[v])continue;
        d[v]=d[u]+e[i].w;
        getdis(v,u);
    }
}
void cal(int u){
    int p=0;//计以u为根的树的所有路径长度数
    for(int i=h[u];i;i=e[i].next){
        int v=e[i].v;
        if(vis[v])continue;
        rem[0]=0;//计子树v的路径长度数
        d[v]=e[i].w;
        getdis(v,u);
        for(int j=rem[0];j;j--){
            for(int k=1;k<=m;k++){
                if(query[k]>=rem[j])
                    ans[k]|=judge[query[k]-rem[j]];//如果距离为 query[k]-rem[j] 的路径存在 query[k]的路径长度也存在 
            }
        }
        for(int j=1;j<=rem[0];j++){
            q[++p]=rem[j];
            judge[rem[j]]=1;//记录路径长度
        }
    }
    //这棵子树搜完了，将这棵子树的所有路径长度清零
    for(int i=1;i<=p;i++)judge[q[i]]=0;//用memset会T!
}
void solve(int u,int S){
    maxsize=S;
    getroot(S,u,0);
    vis[root]=1;
    judge[0]=1;
    cal(root);
    for(int i=h[root];i;i=e[i].next){
        int v=e[i].v;
        if(vis[v])continue;
        solve(v,sz[v]);
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
        add(y,x,z);
    }
    for(int i=1;i<=m;i++)
        scanf("%d",&query[i]);
    solve(1,n);
    for(int i=1;i<=m;i++)
        if(ans[i])printf("AYE\n");
        else printf("NAY\n");
    return 0;
}