P3806 【模板】点分治1

题目大意：

求树上有没有距离为k的路径。

思路：

点分治的模板题===
点分治就是每次找到树的重心，然后dfs子树，算到根节点的距离，对于两颗子树，dis[x]+dis[y]==k，k就是存在的，然后把子树出现的值合并。然后删掉跟节点，把每一颗子树做以上操作。看最后有没有k。

程序：

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<queue>
#define N 100000
#define inf 123456789
using namespace std;
queue <int> q;
int n,m,sum,cnt,root,top,pre;
int size[N],dis[N],a[N],vis[N],f[N],g[N],ans[N],last[N],ss[N];
struct edge{int to,next,w;}e[N*2];

void add(int x,int y,int z){
    e[++cnt].to=y; e[cnt].next=last[x]; e[cnt].w=z; last[x]=cnt;
    e[++cnt].to=x; e[cnt].next=last[y]; e[cnt].w=z; last[y]=cnt;
}

void getroot(int x,int fa){
    f[x]=0; size[x]=1;
    int y=0;
    for (int i=last[x];i;i=e[i].next){
        y=e[i].to;
        if (y==fa||vis[y]) continue;
        getroot(y,x);
        size[x]+=size[y];
        f[x]=max(f[x],size[x]);
    }
    f[x]=max(f[x],sum-size[y]);
    if (f[x]<f[root]) root=x;
}

void dfs(int x,int fa){
    size[x]=1;
    for (int t=last[x];t;t=e[t].next){
        int y=e[t].to;
        if (y==fa||vis[y]) continue;
        dis[y]=dis[x]+e[t].w;
        g[++top]=dis[y];
        dfs(y,x);
        size[x]+=size[y];
        if (x==root) {
            for (int i=pre;i<=top;i++){
                for (int j=1;j<=m;j++){
                    if (a[ss[j]-g[i]]) ans[j]=1;
                }
            }
            for (int i=pre;i<=top;i++) a[g[i]]=1;
            pre=top+1;
            q.push(y);
        }
    }
}

int main(){
    scanf("%d%d",&n,&m);
    for (int i=1;i<n;i++){
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
    }
    for (int i=1;i<=m;i++){
        scanf("%d",&ss[i]);

    }
    q.push(1);
    size[1]=n;
    f[0]=inf;
    while (!q.empty()){
        int x=q.front();
        q.pop();
        sum=size[x];
        root=0;
        getroot(x,0);
        dis[root]=0;
        pre=1; top=0; a[0]=1;
        dfs(root,0);
        for (int i=1;i<=top;i++) a[g[i]]=false;
        vis[root]=1; 
    }
    for (int i=1;i<=m;i++){
        if (ans[i]) printf("AYE\n");
            else printf("NAY\n");
    }
}