Luogu P3806 点分治模板1

题意：

　　给定一棵有n个点的树询问树上距离为k的点对是否存在。

分析：

　　这个题的询问和点数都不多（但是显然暴力是不太好过的，即使有人暴力过了）

　　这题应该怎么用点分治呢。显然，一个模板题，我们直接用套路，每次找重心，对于这个重心处理，过当前点的符合要求的路径。

　　我们可以看到这个最大长度1e7，开数组是开得下的，所以维护一个数组（或者bitset也行）来保存之前的子树中到根距离的长度，之后访问到每个子树时，先查询，再记录，最后删除标记，将所有的点都统计完之后，答案便被保存到了数组中，直接按照题意输出即可。

代码：

 #include<bits/stdc++.h>
 using namespace std;
 const int N=,inf=1e7;
 struct node{int y,z,nxt;}e[N<<];
 int n,m,c=,h[N],mx[N],siz[N],d[N],rem[N],sm;
 int t[inf],jd[inf],vis[N],que[N],q[N],rt,ans;
 void add(int x,int y,int z){
     e[++c]=(node){y,z,h[x]};h[x]=c;
     e[++c]=(node){x,z,h[y]};h[y]=c;
 } void getrt(int x,int fa){//负责找重心
     siz[x]=;mx[x]=;
     for(int i=h[x],y;i;i=e[i].nxt)
     if((y=e[i].y)!=fa&&!vis[y]){
         getrt(y,x);siz[x]+=siz[y];
         mx[x]=max(mx[x],siz[y]);
     } mx[x]=max(mx[x],sm-siz[x]);
     if(mx[x]<mx[rt]) rt=x;return ;
 } void dfs(int x,int fa){//负责求点到子树中点的距离
     rem[++rem[]]=d[x];
     for(int i=h[x],y;i;i=e[i].nxt)
     if((y=e[i].y)!=fa&&!vis[x])
     d[y]=d[x]+e[i].z,dfs(y,x);
 } void calc(int x){ int p=;//负责处理x点为根的子树
     for(int i=h[x],y;i;i=e[i].nxt)
     if(!vis[y=e[i].y]){
         rem[]=;d[y]=e[i].z;dfs(y,x);
         for(int j=rem[];j;j--)
         for(int k=;k<=m;k++)
         if(que[k]>=rem[j])
         t[k]|=jd[que[k]-rem[j]];
         for(int j=rem[];j;j--)
         q[++p]=rem[j],jd[rem[j]]=;
     } for(int i=;i<=p;i++) jd[q[i]]=;
 } void solve(int x){//负责每次找重心，层层分治
     vis[x]=jd[]=;calc(x);
     for(int i=h[x],y;i;i=e[i].nxt){
         if(vis[y=e[i].y]) continue;
         sm=siz[y];mx[rt=]=inf;
         getrt(y,);solve(rt);
     } return;
 } int main(){
     scanf("%d%d",&n,&m);
     for(int i=,x,y,z;i<n;i++)
     scanf("%d%d%d",&x,&y,&z),add(x,y,z);
     for(int i=;i<=m;i++) scanf("%d",&que[i]);
     mx[rt]=sm=n;getrt(,);solve(rt);
     for(int i=;i<=m;i++)
     if(t[i]) puts("AYE");else puts("NAY");
     return ;
 }

点分治