【笔记】点分治

点分治是一类分治算法 呸!废话 ，主要解决树上路径权值统计问题

给一道例题：bzoj2152聪聪可可
【题目概要】

给定一棵树和树上的边权，任选两点$x,y$，求$x$到$y$的距离$mod$ $3$为$0$的概率
$1 \leq n \leq 1e5$
$1 \leq m \leq 1e5$

【Solution】

显然不可能直接枚举两个点并直接计算路径，暴力$O(n^3)$，树剖或LCA差分$O(n^2\log_2 n)$显然过不了

考虑点分治，每次选取当前树的重心，只考虑经过当前节点的路径

由于每次选取的是重心进行分割，子树的大小一定不超过当前的一半，
所以可证这么做的时间复杂度是$O(n \log_2 n)$

那么如何统计经过这个点的路径中符合条件的个数呢

从该点进行dfs，求出所有当前统领的子树内所有点到该点的距离，根据统计出的距离计数

比如说这题，统计操作中只用统计距离$mod$ $3$的值的个数即可，最后答案就是同余1和同余2的互搭，同余0的自由搭配

但这样会将在同一棵子树内的路径算进来，那么对每一个儿子再进行一次，答案减去即可

具体见代码：

#include<bits/stdc++.h>
using namespace std;
#define rg register
#define inf (0x7fffffff)
#define cl(x) memset(x,0,sizeof(x))

template <typename _Tp> inline void read(_Tp&x){
    rg char c11=getchar(),ob=0;x=0;
    while(c11!='-'&&!isdigit(c11))c11=getchar();if(c11=='-')c11=getchar(),ob=1;
    while(isdigit(c11))x=x*10+c11-'0',c11=getchar();if(ob)x=-x;return ;
}

const int N=20500;
struct Edge{int v,w,nxt;}a[N<<1];
int head[N],depth[N],cnt[3],root,f[N],sz[N];char vis[N];
int ans=0,n,sn,_=0;

inline void add(int u,int v,int w){a[++_].v=v,a[_].w=w,a[_].nxt=head[u],head[u]=_;return ;}

inline int gcd(int x,int y){return y?gcd(y,x%y):x;}

inline void get_root(int x,int fa){
    sz[x]=1,f[x]=0;
    for(rg int i=head[x];i;i=a[i].nxt)
        if(a[i].v!=fa&&!vis[a[i].v])
            get_root(a[i].v,x),sz[x]+=sz[a[i].v],f[x]=max(f[x],sz[a[i].v]);
    f[x]=max(f[x],sn-sz[x]);
    if(f[root]>f[x])root=x;
    return ;
}

inline void get_deep(int x,int fa){
    ++cnt[depth[x]%3];
    for(rg int i=head[x];i;i=a[i].nxt)
        if(!vis[a[i].v]&&a[i].v!=fa)
            depth[a[i].v]=depth[x]+a[i].w,get_deep(a[i].v,x);
    return ;
}

inline int calc(int x){
    cnt[0]=cnt[1]=cnt[2]=0;get_deep(x,0);
    return cnt[0]*cnt[0]+cnt[1]*cnt[2]*2;
}

inline void work(int x){
    vis[x]=1,depth[x]=0;
    ans+=calc(x);
    for(rg int i=head[x];i;i=a[i].nxt)
        if(!vis[a[i].v])
            depth[a[i].v]=a[i].w,ans-=calc(a[i].v),sn=sz[a[i].v],root=0,get_root(a[i].v,0),work(root);
    return ;
}

int main(){
    read(n);
    rg int x,y,z;
    for(rg int i=1;i<n;++i)read(x),read(y),read(z),add(x,y,z),add(y,x,z);
    f[0]=inf;
    sn=n,root=0;get_root(1,0);
    work(root);
    x=ans,y=n*n;z=gcd(x,y);
    printf("%d/%d\n",x/z,y/z);
    return 0;
}

其实大多数的点分治的题目都是大同小异，只用改一下$calc()$函数和$get\_deep()$就行

但有一些题目可能需要一点技巧，比如说：
bzoj2599，LG4149，IOI2011-Race就需要用桶进行优化
bzoj3697需要分类讨论