点分治,是一种处理树上路径问题的工具。
举例:给定一棵树和一个整数 k ,求树上等于 k 的路径有多少条?
1、找重心(重心所有的子树的大小都不超过整个树大小的一半)是最优的
树形DP
Size=n;
void get_root(int u,int fa)
{
f[u]=0,size[u]=1;//f数组代表这个点最大子树的大小,size数组代表子树大小的和
for(int i=head[u];i;i=next[i])
{
int v=to[i];
if(use[v]||v==fa)continue;//use表示已经遍历过了 ,但此处无用
getroot(v,u);
f[u]=max(f[u],size[v]);
size[u]+=size[v];
}
f[x]=max(f[x],Size-size[x]);//容斥原理,还有上面那一部分
if(f[x]<f[root])root=x;//重心的深度要尽可能小
}
2、求距离
先dfs一遍,求重心到各个节点的距离
再枚举:
如果两点到重心的距离和为k 或 某点到重心的距离为k,辣么答案加1
3、统计答案
枚举一个点,另一个点二分求解:sort排序,相同距离的点是连续的,所以可找出l与r,ans+=r-l+1
但是,在同一子树中,到重心的路径有重叠部分
处理方法:
1.求距离时把点染色,不同子树不同颜色,求答案时就枚举每个符合答案的每个点看是否不在一个子树里
2.可以求当前点儿子的答案,统计儿子答案时各个点的距离加上儿子到根的距离,即把符合在一个子树条件的情况统计出来,最后这个点的答案减去儿子答案就行了