Prime Distance On Tree CodeChef - PRIMEDST(点分治+FFT)

题目链接

题意：输出一棵树上所有的路径长度为素数的链的个数

思路：树上的路径问题比较容易想到的是用点分治来解决。

1、找到树的重心（重心就是最大子树最小的节点）

2、从重心开始遍历子树

3、将当前子树所有节点距离重心的长度暂时保存起来

4、将暂时保存的路径长度 与 之前的所有路径长度进行一一匹配得到当前子树对于答案的一部分贡献

5、另一部分贡献是与重心之间的距离

6、将暂时保存的路径加入所有的路径长度

7、遍历下一个子树回到步骤2，知道重心的所有子树遍历完成

8、将重心下沉到所有子树中去，回到步骤1

9、最后就可以得到所有的长度的个数，素数筛一遍之后用素数答案除以总答案就可以了

以这样的步骤就可以得到所有的路径的情况，但是在路径之间一一匹配的时候考虑到复杂度是 $O(n^2)$ 的，这个步骤需要优化一下。

于是就可以想到用多项式乘法来表示匹配过程，用参数的指数表示路径长度，系数相乘代表情况之间的匹配，这样指数就会相加。这个过程用FFT来做的话复杂度可以降低到 $nlogn$ ，这样这个题理论就可以做了。

然后需要注意的就是多项式数组的清空。

上代码

#include <bits/stdc++.h>
using namespace std;
const int N = 100004;
const double PI = acos(-1.0);
int n, minn, sz[N], mx[N], vis[N], rt, cnt[N];
vector <int> g[N];

const int M = 1e7;
int isprime[M+10], prime[M+10];
int pn = 0;

void getprime() {
   
	memset(isprime, -1, sizeof(isprime));
	isprime[1] = 0;
	for(int i = 2;i < M;i++){
   
		if(isprime[i]) prime[pn++] = i;
		for(int j = 0;j < pn && 1ll*i*prime[j] < M;j++) {
   
			isprime[i*prime[j]] = 0;
			if(i % prime[j] == 0) break;