bzoj 1468: Tree 点分治

Description

给你一棵TREE,以及这棵树上边的距离.问有多少对点它们两者间的距离小于等于K

Input

N（n<=40000） 接下来n-1行边描述管道，按照题目中写的输入 接下来是k

Output

一行，有多少对点之间的距离小于等于k

题解：

十分巧妙的一种数据结构
大致是这样的：
首先考虑在原树中如何解决，再考虑当树的深度过大的时候会 TLE，于是我们就用点分治来优化一下.
假如当前遍历到点 $x$，先遍历一下 $x$ 为根的子树中产生的所有贡献.
然而，贡献会有不合法情况（来自 $x$ 在原树中同一个儿子即为不合法）.
再依次遍历一下每一个儿子，剪掉不合法情况.
但是........
下一步，我们访问的不是点 $x$ 再原树中的所有儿子，而是 $x$ 再原树中所有儿子为根的子树中的重心！！！！
这样，我们就可以保证，我们所遍历的树的深度是 $O(logn)$ 级别的.
这是点分治的一种，还有一种方法就是在统计答案的时候有差异.
即每次遍历 $x$ 在原树中的每一个儿子为根的子树，将子树两两合并，这样不会有不合法的情况出现，但是不太好写(比第一种方法少一个 $log$)

Code:

#include<bits/stdc++.h>
#define N 400010 
#define setIO(s) freopen(s".in","r",stdin) 
using namespace std;
int to[N<<1],head[N<<1],nex[N<<1],val[N<<1];
int f[N],root,m,dep[N],siz[N],sn,d[N],tot,ans,cnt; 
int vis[N];
void add(int u,int v,int c){ nex[++cnt]=head[u],head[u]=cnt,to[cnt]=v,val[cnt]=c;  }
//求重心
void get(int x,int fa){
    f[x]=0,siz[x]=1; 
    for(int v=head[x];v;v=nex[v]){
        if(to[v]==fa||vis[to[v]]) continue; 
        get(to[v],x); 
        siz[x]+=siz[to[v]]; 
        f[x]=max(f[x],siz[to[v]]); 
    }      
    f[x]=max(f[x],sn-siz[x]); 
    if(f[x]<f[root]) root=x; 
}
void getdis(int x,int fa){
    d[++tot]=dep[x];
    for(int v=head[x];v;v=nex[v]){
        if(to[v]==fa||vis[to[v]]) continue; 
        dep[to[v]]=dep[x]+val[v],getdis(to[v],x); 
    }
}
int calc(int x){     
    tot=0; 
    getdis(x,0); 
    sort(d+1,d+tot+1);  
    int i=1,j=tot,sum=0; 
    while(i<j) {
        if(d[i]+d[j]<=m) {
            sum+=j-i,i++;  
        }
        else j--;              
    }
    return sum; 
}
void dfs(int x){
    dep[x]=0; 
    vis[x]=1; 
    ans+=calc(x); 
    for(int v=head[x];v;v=nex[v]){
        if(!vis[to[v]]) {
            dep[to[v]]=val[v]; 
            ans-=calc(to[v]);          
            sn=siz[to[v]]; 
            root=0; 
            get(to[v],0); 
            dfs(root); 
        }
    }
}
int main(){
    // setIO("input"); 
    int n; 
    scanf("%d",&n); 
    for(int i=1;i<n;++i) 
    {
        int x,y,z; 
        scanf("%d%d%d",&x,&y,&z); 
        add(x,y,z),add(y,x,z); 
    }
    cnt=0,ans=0; 
    scanf("%d",&m); 
    f[0]=0x7f7f7f7f; 
    sn=n; 
    root=0,get(1,0),dfs(root);
    printf("%d",ans); 
    return 0; 
}