写在前面
先开个坑...
之前学过点分治,但是总是感觉打的时候内心莫名的慌,敲完几个函数的定义就开始脑袋一片空白。
所以重学了一下点分治,并写了这篇博客。
看看什么时候把它补完吧。
\(\mathrm{1}\) 参考资料
找了两篇还不错的博客
https://www.cnblogs.com/bztMinamoto/p/9489473.html
https://www.cnblogs.com/PinkRabbit/p/8593080.html
\(\mathrm{2}\) 问题提出
考虑这样一个问题:给出一棵树,求有多少个点对 \((x,y)\) 满足它们之间的距离不超过 \(k\) / 等于 \(k\) / 大于 \(k\)。
\(1 \le n \le 10^5\)
\(\mathrm{2.1}\) 暴力解法一
很显然,求两个点点对数目,可以枚举这两个点,然后求它们的距离。
时间复杂度 \(O(n^3)\) ,如果命题人给了这个复杂度超过 \(20pts\) 可真是良心。
\(\mathrm{2.2}\) 暴力解法二
可以枚举起点 \(x\) ,然后遍历这棵树,一旦距离超过 \(k\) 就返回。
如果是求大于 \(k\) 的,就先求出小于等于 \(k\) 的,然后容斥。
时间复杂度 \(O(n^2)\) ,正常的命题人应该会给到 \(30pts-60pts\) 。
\(\mathrm{2.3}\) 正解
点分治。
\(\mathrm{3}\) 点分治
\(\mathrm{3.1}\) 点分治思想和理解
点分治,顾名思义,是一种与点有关的分治算法。
分治分治,分而治之。树形结构是一种有序的结构,其有一种自然的,从上往下的秩序,分治自然按照这样的顺序进行。
但是如果我们就以 \(1\) 为根,可以被菊花图卡到 \(O(n^2)\) ?所以我们不能以 \(1\) 为根。
但是选择的这个根肯定不能是随机的不然得分也是随机的。考虑树上的特殊的点,自然想到树的重心。树的重心保证了删掉它之后,最大的子树不超过 \(\frac{n}{2}\) ,所以复杂度上界是 \(O(n log n)\) 的。
因此,点分治的代码中,自然有 getroot
函数。
我们称当前选择的这个重心为 当前的分治中心
每次的分治中心分治完后,我们继续向它的子树分治,因此每一个 分治中心 会把当前的 分治区域 分为若干个 新的更小的分治区域 ,这就是大规模问题向类似的小规模问题转化的过程,也符合点分治名字中 分治 的意义。
\(\mathrm{3.2}\) 点分治实现
首先给出点分治的流程:
- 求树的重心
- 以树的重心为分治中心为根开始点分治
- 寻找当前分治块的重心,作为分治中心
- 删除当前分治中心,计算当前分治中心答案的贡献,删除重复贡献。
- 在删除当前分治中心产生的联通块中继续分治,重复3-5,直到无法分治
流程代码:
void dfs(int x){
ans+=calc(x,0);del[x]=1;
for(int i=Head[x];i;i=Next[i]){
int y=to[i];
if(del[y]) continue;
ans-=calc(y,w[i]);
sz=size[y],mx=INF;
getroot(y,0);dfs(root);
}
}
其中比较难以理解的语句为
ans-=calc(y,w[i]);
将在后面的博文中解释。
上述代码中,calc
为计算答案的代码,因题而异, del
表示删除当前分治中心,getroot
为寻找重心, sz
为当前分治块的大小。
其中 size
等需要预处理。
\(\mathrm{3.3}\) 删除重复贡献
对于红色的这样一条路径,从 \(y\) 的某个子树中经过 \(y\) ,进入 \(y\) 的另外一个子树的一条有贡献的路径,在 \(x\) 的时候可能会:
重复经过一条边,导致被多次计算。
\(\mathrm{4}\) 一些例题
POJ