题目描述
题解
考虑到恰好为 i i i 不好求,我们可以求 i i i 的倍数的个数然后容斥。于是我们可以枚举 x x x 作为 lca \text{lca} lca ,我们要计算 d ( u , x ) , d ( v , x ) d(u,x),d(v,x) d(u,x),d(v,x) 都是 i i i 的倍数的 ( u , v ) (u,v) (u,v) 数对数。因为这个跟深度有关,容易想到长链剖分。但是我们不能去跳最深深度的倍数,所以我们考虑分块,如果 d e e p > n deep>\sqrt n deep>n ,那么单次跳不会超过 n \sqrt n n 次,否则我们可以设 g [ i ] [ j ] g[i][j] g[i][j] 表示 d p [ v ] % i = j dp[v] \% i=j dp[v]%i=j 的 v v v 的个数,枚举 i i i 即可。这样就避免了枚举最深深度,因此总效率为 O ( n n ) O(n \sqrt n) O(nn) 。