【题解】USB 的数学题

求 $G(n)={\sum }_{i=1}^n\frac{i}{gcd(i,n)}$ 的值，其中 $gcd(i,n)$ 为 $i$ 和 $n$ 的最大公约数。

输入数据以多组数据给出，数据组数不超过 $10^5$，$1\le n\le 10^6$。

---

我们先考虑按 $i=1...n$ 中的 $gcd(i,n)$ 分类，当 $n=6$ 时，分类如下图：

因此，对于上式的 $G(n)$ 也可以写成：$G(n)={\sum }_{d|n}^n\frac{d}{n}\phi (\frac{d}{n})$，即 $G(n)={\sum }_{d|n}^{n}d\phi (d)$。

定理：$G(n)={\sum }_{d|n}^{n}d\phi (d)$ 为积性函数。

证明：设有正整数 $n,m$ 且互质，所以 $n,m$ 的约数互不相同（不然最大公因数就不是 $1$），所以 $nm$ 的约数个数为 $n,m$ 约数个数的积。因为欧拉函数本身是积性函数，所以可得以下结论：

$$G(n)G(m)=\sum _{i|n}^n\sum _{j|m}^{m}ij\phi (i)\phi (j)=\sum _{i|n}^n\sum _{j|m}^{m}ij\phi (ij)=G(nm)$$

其中枚举 $i,j$ 后的 $ij$ 项已经枚举了 $nm$ 的所有约数，因此 $G(n)$ 为积性函数。

众所周知，所有积性函数都可以用线性筛法递推得到。对于任意一个质数 $x$ 来说，$G(x)=(x-1)x+1$，这很容易通过式子证明。

当递推 $i\ast prime[j]=k$ 时，若 $prime[j]$ 不是 $i$ 的质因子，那么直接有 $G(k)=G(i)G(prime[j])$。

若 $prime[j]$ 是 $i$ 的质因子，分两种情况：

1. $i$ 不是 $prime[j]$ 为底数的幂。
   我们从 $i$ 中提取 $i$ 最小质因子的指数 $mi[i]$ 和幂 $highPow[i]$。由线性筛的原理得知，$prime[j]$ 也是 $i$ 的最小质因子。令 $A=highPow[i]\ast prime[j],B=i\ast prime[j]/A$，那么很容易递推 $G(i\ast prime[j])=G(A)G(B)$。

2. $i$ 是 $prime[j]$ 为底数的幂。
   若用上述方法，会得出 $A=i\ast prime[j],B=1$，无法递推。设 $k=i\ast prime[j]$，利用之前的式子，很容易可知 $G(k)={\sum }_{p=0}^{mi[k]}prime[j{]}^p\phi (prime[j{]}^p)$。由欧拉函数的计算公式可知，$\phi (prime[j{]}^p)=prime[j{]}^p(1-\frac{1}{prime[j]})$。因此，上述 $G(k)$ 的公式可以用暴力实现。

其中 $highPow[]$ 和 $mi[]$ 都可以通过线性筛来实现。如果是这样，我们就可以不用计算欧拉函数了。

---

#include<bits/stdc++.h>
#define int long long
#pragma GCC optimize(1)
#pragma GCC optimize(2)
#pragma GCC optimize(3,"Ofast","inline")
using namespace std;
int n,t,x,prime[10000010],m,ans,g[10000010],highPow[10000010],mi[10000010],w[2][10000010];
bool b[10000010];
signed main()
{
	scanf("%lld",&t);
	n=10000000;
	for(int i=2;i<=n;i++)
	{
		if(!b[i]) //没有被筛选过，说明 i 是质数
		{
			g[i]=i*(i-1)+1;
			highPow[i]=i;
			mi[i]=1;
			prime[++m]=i;
		}
		for(int j=1;j<=m&&i*prime[j]<=n;j++)
		{
			int x=i*prime[j]; //k=x=i*prime[j]
			b[x]=true; //筛选质数
			if(i%prime[j]) //i 和 prime[j] 互质
			{
				int A=i,B=prime[j];
				g[x]=g[A]*g[B];
				highPow[x]=B;
				mi[x]=1;
			}
			else
			{
				highPow[x]=highPow[i]*prime[j];
				mi[x]=mi[i]+1;
				int A=highPow[x];
				int B=i*prime[j]/A;
				if(B==1) //分情况，如果 i 是 prime[j] 的幂
				{
					w[0][0]=w[1][0]=1;
					w[0][1]=prime[j];
					w[1][1]=prime[j]-1;
					for(int k=2;k<=mi[x];k++)
					{
						w[0][k]=w[0][k-1]*prime[j];
						w[1][k]=w[1][k-1]*prime[j];
					}
					for(int k=0;k<=mi[x];k++)
					{
						g[x]+=w[0][k]*w[1][k];
					}
				}
				else
				{
					g[x]=g[A]*g[B];
				}
				break; 
			}
		}
	}
	while(t--)
	{
		scanf("%lld",&x);
		if(x-1)printf("%lld\n",g[x]); //if(x-1)就是x不等于1
		else printf("1\n");//特判 1 的情况
	}
	return 0;
}