luogu P1891 疯狂LCM

背景：

推了好久的式子，发现一开始错了。

题目传送门：

https://www.luogu.org/problem/P1891


$$\sum _{i=1}^n\text{lcm}(i,n)$$

$$=\sum _{i=1}^n\frac{n\cdot i}{\gcd (i,n)}$$

$$=n\sum _{k=1}^n\sum _{i=1}^n\frac{i}{k}[\gcd (i,n)=k]$$

显然有$k|n$，所以：
$$=n\sum _{k|n}\sum _{i=1}^n\frac{i}{k}[\gcd (i,n)=k]$$

$$=n\sum _{k|n}\sum _{i=1}^{\frac{n}{k}}\frac{i}{k}[\gcd (\frac{i}{k},\frac{n}{k})=1]$$

$$=n\sum _{k|n}\sum _{i=1}^{k}i[\gcd (i,\frac{n}{k})=1]$$

令$d=\frac{n}{k}$，有：
$$=n\sum _{d|n}\sum _{i=1}^{d}i[\gcd (i,n)=1]$$

考虑${\sum }_{i=1}^{d}i[\gcd (i,n)=1]=\frac{d\cdot {\varphi }_d}{2}$，$\Theta (n\log n)$预处理即可。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;
	int n;
	int prime[1000010],phi[1000010];
	LL ans[1000010];
void init(int ma)
{
	int t=0;
	phi[1]=1;
	for(int i=2;i<=ma;i++)
	{
		if(!phi[i]) prime[++t]=i,phi[i]=i-1;
		for(int j=1;j<=t&&i*prime[j]<=ma;j++)
		{
			if(!(i%prime[j]))
			{
				phi[i*prime[j]]=phi[i]*prime[j];
				break;
			}
			phi[i*prime[j]]=phi[i]*phi[prime[j]];
		}
	}
	for(int i=1;i<=ma;i++)
		for(int j=1;i*j<=ma;j++)
			ans[i*j]+=(((LL)phi[j]*j+1)>>1);
}
int main()
{
	int T;
	init(1000000);
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		printf("%lld\n",ans[n]*n);
	}
}