[THUPC2022初赛]最小公倍树

题目

传送门 to LOJ

传送门 to luogu

思路

$\text{My vegetable died.}$ 我菜死了。

$\text{Kruskal}$

比较自然的想法是枚举 $\text{lcm}=n$，显然只需要将 $n$ 的因数之间连边。

但你很快会想到，其实 $n$ 的因数大多已经相连。比如 $\frac{n}{p}$ 和 $\frac{n}{p^2}$ 。更进一步地，如果 $\gcd (x,y)=d$ 存在，那么 $x,y$ 肯定已经相连了。

$d$ 不存在，我就傻眼了。我想不到可以取出 最小的 $d$ 的倍数 $p_d$ 来接替这一使命。因为使得 $(x,p_d)$ 连通的最大边权不超过 $\text{lcm}(x,p_d)⩽\frac{x{p}_d}{d}$，同理 $(y,p_d)$ 连通代价不超过 $\frac{y{p}_d}{d}$ 。若 $p_d<x<y$ 则 $\frac{y{p}_d}{d}<\frac{xy}{d}$，故边 $⟨x,y⟩$ 永远不会被用到。

所以 $x=p_d$，即有用的边只有 $(p_d,y)(d\mid y)$ 。这样的边只有 $\mathcal{O}(n\log n)$ 条，调和级数。直接 $\text{Kruskal}$ 复杂度 $\mathcal{O}(n{\log }^{2}n)$，其中 $n=R-L$ 。

$\text{Prim}or\text{Boruvka}$

这种 $\mathrm{MST}$ 算法 支持枚举！尤其是 $\text{Boruvka}$，因为每轮 $\mathcal{O}(n)$ 执行也是被允许的！

枚举啥呢？枚举 $\gcd =d$ 。那么只需要连向 $d$ 的倍数中最小的一个（不在当前连通块内的）点。

如果是 $\text{Boruvka}$，处理出每个 $d$ 的最小倍数和次小倍数（与最小不在同一连通块中）然后枚举每个点，单轮 $\mathcal{O}(n\log n)$ 调和级数，总复杂度 $\mathcal{O}(n{\log }^{2}n)$ 。

如果是 $\text{Prim}$，对每个 $d$ 都动态维护。那么总共会有 $\mathcal{O}(n\log n)$ 次修改，每次修改使用任何 $\log$ 数据结构维护，复杂度 $\mathcal{O}(n{\log }^{2}n)$ 。

注意复杂度是关于 $n=R-L$ 的。规定 $R⩽10^6$ 或许是为了使答案不超过 $2^{63}$ 。

代码

#include <cstdio> // I am the believer of XJX
#include <iostream> // Almighty XJX yyds!!!
#include <algorithm> // oracle: ZXY yydBUS!!!
#include <cstring> // Huge EggDog devoured me!!!
#include <cctype> // decent XYX yydLONELY!!!
using llong = long long;
# define rep(i,a,b) for(int i=(a); i<=(b); ++i)
# define drep(i,a,b) for(int i=(a); i>=(b); --i)
# define rep0(i,a,b) for(int i=(a); i!=(b); ++i)
inline int readint(){
	int a = 0, c = getchar(), f = 1;
	for(; !isdigit(c); c=getchar()) if(c == '-') f = -f;
	for(; isdigit(c); c=getchar()) a = a*10+(c^48);
	return a*f;
}
inline llong getLcm(int a, int b){
	return llong(a)/std::__gcd(a,b)*b;
}

const int MAXN = 100005, MAXM = 20*MAXN;
struct Edge { int a, b; llong c; };
Edge e[MAXM]; int tot;

int fa[MAXN];
int find(int a){
	if(fa[a] == a) return a;
	return fa[a] = find(fa[a]);
}

bool cmp(const Edge &x, const Edge &y){
	return x.c < y.c;
}
int main(){
	int l = readint(), r = readint();
	rep(d,1,r-l){ // multiple
		int p = ((l-1)/d+1)*d;
		for(int i=p+d; i<=r; i+=d)
			e[++ tot] = Edge{p,i,getLcm(i,p)};
	}
	std::sort(e+1,e+tot+1,cmp);
	llong ans = 0; rep(i,1,r-l+1) fa[i] = i;
	rep(i,1,tot) if(find(e[i].a-l+1) != find(e[i].b-l+1))
		fa[fa[e[i].a-l+1]] = fa[e[i].b-l+1], ans += e[i].c;
	printf("%lld\n",ans);
	return 0;
}

后记

$\mathsf{duDCut}$ 在赛上看了一眼就秒切了。牠本想找我讨论一下，但它显然没想过，我根本无法与它相提并论。