点分治学习笔记（1）

给定一棵有n个点的树
询问树上距离为k的点对是否存在。
输入格式：
n,m 接下来n-1条边a,b,c描述a到b有一条长度为c的路径
接下来m行每行询问一个K
输出格式：
对于每个K每行输出一个答案，存在输出“AYE”,否则输出”NAY”(不包含引号)

朴素的想法就是直接去枚举。但我们可以发现（我发现不了）。
对于一个点 显然只有经过他的路径和不经过他的路径。
我们不考虑不过他的路径。经过他的路径的两个端点一定他在的两个子树里。
以他为根统计到他子树节点的距离。这样的单独的距离或两段距离之和一定经过这个根节点。

对于每个点我们都这样计算就好辣。
可是问题来了，当树退化成一条链的时候。复杂度就会很高，不断的递归找子树。
所以我们需要按树的重心（因为重心删掉后子树最小）来找这些点，这样可以把复杂度控制在（nlogn）。

所以点分治的思想就是——
1.找重心把它作为根
2.解决根的路径问题
3.递归子树解决子问题

神犇博客讲的特别好：http://blog.csdn.net/ALPS233/article/details/51398629

#include<bits/stdc++.h>
using namespace std;

const int MAXN=1e5+5;

struct edge{
    int to,next,w;
}e[MAXN<<1];

struct seg{
    int len,pos;
}seg1[MAXN<<1];

int head[MAXN],cnt=0,cnt2=0;
inline void add(int u,int v,int w){e[++cnt]=(edge){v,head[u],w},head[u]=cnt;}
inline void add2(int dis,int pos){seg1[++cnt2]=(seg){dis,pos};}

int f[MAXN],sum,vis[MAXN],size[MAXN],rt,dis[MAXN];
bool ok[10000001];

void dfs1(int u,int fa){//找重心成为新的根。 
    size[u]=1;
    f[u]=0;
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to;
        if(v==fa||vis[v])continue;
        dfs1(v,u);
        size[u]+=size[v];
        f[u]=max(f[u],size[v]);
    }
    f[u]=max(f[u],sum-size[u]);
    if(f[u]<f[rt])rt=u;
}

void dfs2(int u,int fa,int num){
    size[u]=1;
    add2(dis[u],num);
    ok[dis[u]]=1;
        //进行一些操作 记录dis【u】之类的 
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to,w=e[i].w;
        if(vis[v]||v==fa)continue;
        dis[v]=dis[u]+w;
        dfs2(v,u,num);
        size[u]+=size[v];
    }
}

void cal(int u){//是用来处理你所需信息的,计算经过u的路径，在其中dfs2 
    int temnum=0;cnt2=0;
        //进行一些初始化之类的 
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to,w=e[i].w;
        if(vis[v])continue;
        dis[v]=w;
        dfs2(v,u,++temnum);
    }
}

void solve(int u){
    vis[u]=1;//删点 
        cal(u);//处理信息 那题要的东西 
    //进行一些统计操作 
    for(int i=1;i<cnt2;i++)
        for(int j=i+1;j<=cnt2;j++)
            if(seg1[i].pos!=seg1[j].pos)
                ok[seg1[i].len+seg1[j].len]=1;  
    for(int i=head[u];i;i=e[i].next){
        int v=e[i].to;
        if(vis[v])continue;
        f[0]=sum=size[v];//重置总点数 总联通块数 
        dfs1(v,rt=0);//找子树的根 
        solve(rt);//递归解决 子树问题 
    }
}

int n,q;

int main(){ 
    int tem1,tem2,tem3;
    memset(ok,0,sizeof(ok));
    memset(vis,0,sizeof(vis));
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n-1;i++){
        scanf("%d%d%d",&tem1,&tem2,&tem3);
        add(tem1,tem2,tem3);
        add(tem2,tem1,tem3);
    }
    sum=f[0]=n;
    dfs1(1,0);
    solve(rt);
    for(int i=1;i<=q;i++){
        scanf("%d",&tem1);
        if(ok[tem1])printf("AYE\n");
        else printf("NAY\n");
    }
    return 0;
}