洛谷P2303 Longge的问题

题目链接

题意：

给定一个整数$n$，让我们求出表达式${\sum }_{i=1}^{n}gcd(i,n)$的值。

题解：

首先，我们可以发现这个最大公约数一定是$n$的约数。因此我们可以进行分类，也就是枚举$n$的约数。然后原来的表达式就可以改写成：${\sum }_{d|n}d\ast$最大公约数为d的个数，也就是有多少个$x$满足$gcd(x,n)==d$。这个个数该怎么求？
我们发现，如果有$gcd(x,n)==d$，那么$gcd(\frac{x}{d},\frac{n}{d})==1$。 其实这个个数也就是$\varphi (\frac{n}{d})$。

于是上面的表达式就是${\sum }_{d|n}d\ast \varphi (\frac{n}{d})$。

推出上面这个公式，已经可以进行求解了。时间复杂度为$O(1600\sqrt{n})$

不过时间复杂度可以进一步优化到$O(\sqrt{n})$。

我们对上面的表达式进一步化简：

${\sum }_{d|n}d\ast \frac{n}{d}\ast {\prod }_{i=1}^{k_d}(1-\frac{1}{p_i})=n\ast {\sum }_{d|n}{\prod }_{i=1}^{k_d}(1-\frac{1}{p_i})$。

怎么求这个积呢？

假设$n=p_1^{{\alpha }_1}{p}_2^{{\alpha }_2}\dots p_k^{{\alpha }_k}$，那么它的一个因子可以表示为：
$d=p_1^{{\beta }_1}{p}_2^{{\beta }_2}\dots p_k^{{\beta }_k},0\le {\beta }_i\le {\alpha }_i$

那么我们可以对每个质因子依次判断，那么所有的约数可以表示出来：
$(p_1^0+p_1^1+\cdots +p_1^{{\alpha }_1})(p_2^0+p_2^1+\cdots +p_2^{{\alpha }_2})\dots (p_k^0+p_k^1+\cdots +p_k^{{\alpha }_k})$

那么$n$的所有约数都可以从上面每个括号中选择一项组成。

那么这个${\sum }_{d|n}{\prod }_{i=1}^{k_d}(1-\frac{1}{p_i})$表达式的值可以根据每个质因子的选择情况求出，也就是：
$$\begin{gathered} (1+(1-\frac{1}{p_1})+\cdots +(1-\frac{1}{p_1}))\ast (1+(1-\frac{1}{p_2})+\cdots +(1-\frac{1}{p_2}))\ast \cdots = \\ {\sum }_{i=1}^k(1+{\alpha }_i(1-\frac{1}{p_i})) \end{gathered}$$

这样这道题目的就可以在$O(\sqrt{n})$解决。

实现细节见代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int main() 
{
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	ll n;
	cin >> n;
	ll ans = n;
	for (ll i = 2; i * i <= n; i++) {
		ll now = 0, p = i;
		if (n % p == 0) {
			while (n % p == 0) {
				n /= p;
				now++;
			}
			ans /= p;
			ans *= (p + now * p - now);
		}
	}
	if (n > 1) {
		ans /= n;
		ans *= (n + n - 1);
	}
	cout << ans << endl;
    return 0;
}