[Luogu P3806] 点分治 模板

题目描述：

给出一颗含N个点的数。
Q个询问，每个询问一个K，问树中是否存在一对点的距离==K

题目分析：

这道题目可以看为POJ那道的简化版？
然而用那种方法，对于每个K都查一次，不T才怪嘞.
注意到 K<=10000000.
那我们可以用点分治的办法跑一边树，求出所有点对的路径长度，扔进一个桶里即可做到O(1)查询.

题目链接：

Luogu 3806

AC 代码：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
const int maxm=10010;
const int inf=10000000;
int root=0,f[maxm],son[maxm],deep[maxm],d[maxm];
int head[maxm],to[maxm*2],net[maxm*2],cost[maxm*2],cnt,tot,k;
bool vis[maxm];
int ok[inf];
int sum,ans,n,q;
int flag=0;
inline void add(int x,int y,int c){cnt++,to[cnt]=y,cost[cnt]=c,net[cnt]=head[x],head[x]=cnt;}
void getroot(int now,int fa)
{
    son[now]=1,f[now]=0;
    for(int i=head[now];i;i=net[i])
        if(to[i]!=fa&&!vis[to[i]])
        {
            getroot(to[i],now);
            son[now]+=son[to[i]];
            f[now]=std::max(f[now],son[to[i]]);
        }
    f[now]=std::max(f[now],sum-son[now]);
    if(f[now]<f[root]||!root) root=now;
}
void getdeep(int now,int fa)
{
    d[++tot]=deep[now];
    for(int i=head[now];i;i=net[i])
     if(!vis[to[i]]&&to[i]!=fa)
      deep[to[i]]=cost[i]+deep[now],getdeep(to[i],now);
}
int calc(int x,int c,int f)
{
    deep[x]=c;
    tot=0;
    getdeep(x,0);
    //puts("");   
    for(int i=1;i<=tot;i++)
     for(int j=i+1;j<=tot;j++)
     if(d[i]+d[j]<=inf)
     {
        ok[d[i]+d[j]]+=f;
        //printf("%d %d %d\n",d[i],d[j],d[i]+d[j]);
     }

}
void slove(int now)
{
    //printf("-%d-\n",now);
    vis[now]=1;
    calc(now,0,1);
    for(int i=head[now];i;i=net[i])
    if(!vis[to[i]]) 
     calc(to[i],cost[i],-1),sum=son[to[i]],root=0,getroot(to[i],0),slove(root);
}
int main()
{
    //freopen("ts.in","r",stdin);
    //freopen("x2.out","w",stdout);
    scanf("%d%d",&n,&q);
    for(int i=1;i<n;i++)
    {
        int u,v,c;
        scanf("%d%d%d",&u,&v,&c);
        add(u,v,c);
        add(v,u,c);
    }
    sum=n;
    root=0;
    getroot(1,0);
    slove(root);
    for(int i=1;i<=q;i++)
    {
        int k;
        scanf("%d",&k);
        if(ok[k]) puts("AYE");
        else puts("NAY");
    }
    return 0;
}