洛谷 P3768 简单的数学题 题解

题目传送门

题目大意： 求$({\sum }_{i=1}^n{\sum }_{j=1}^{n}ij\gcd (i,j))\mathrm{m}\mathrm{o}\mathrm{d}p$

题解

一看到 $\gcd$ 果断莫比乌斯反演，大力推柿子：
$$\begin{gathered} \sum _{i=1}^n\sum _{j=1}^{n}ij\gcd (i,j) \\ =\sum _{i=1}^n\sum _{j=1}^{n}ij\sum _{k=1}^{n}k[\gcd (i,j)=k] \\ =\sum _{k=1}^{n}k\times k^2\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{n}{k}\rfloor }ij[\gcd (i,j)=1] \\ =\sum _{k=1}^n{k}^3\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{n}{k}\rfloor }ij\sum _{d|\gcd (i,j)}\mu (d) \\ =\sum _{k=1}^n{k}^3\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{n}{k}\rfloor }ij\sum _{d=1}^{\lfloor \frac{n}{k}\rfloor }\mu (d)[d|\gcd (i,j)] \\ =\sum _{k=1}^n{k}^3\sum _{d=1}^{\lfloor \frac{n}{k}\rfloor }\mu (d)\sum _{i=1}^{\lfloor \frac{n}{k}\rfloor }\sum _{j=1}^{\lfloor \frac{n}{k}\rfloor }ij[d|\gcd (i,j)] \\ =\sum _{k=1}^n{k}^3\sum _{d=1}^{\lfloor \frac{n}{k}\rfloor }\mu (d)\times d^2\sum _{i=1}^{\lfloor \frac{n}{kd}\rfloor }\sum _{j=1}^{\lfloor \frac{n}{kd}\rfloor }ij \\ =\sum _{k=1}^n{k}^3\sum _{d=1}^{\lfloor \frac{n}{k}\rfloor }\mu (d)\times d^2\sum _{i=1}^{\lfloor \frac{n}{kd}\rfloor }i\times \sum _{j=1}^{\lfloor \frac{n}{kd}\rfloor }j \end{gathered}$$

设${\sum }_{i=1}^{n}i$ 为$p(n)$，显然 $p(n)=\frac{(1+n)\times n}{2}$那么有：
$$=\sum _{k=1}^n{k}^2\sum _{d=1}^{\lfloor \frac{n}{k}\rfloor }\mu (d)\times d^2\times p(\lfloor \frac{n}{kd}\rfloor {)}^2$$

发现 $p(\lfloor \frac{n}{kd}\rfloor {)}^2$ 里面的这个 $kd$ 比较丑，用 $i$ 代替掉，更换枚举对象。
$$\begin{gathered} =\sum _{i=1}^{n}p(\lfloor \frac{n}{i}\rfloor {)}^2\sum _{k|i}{k}^3\times \mu (\frac{i}{k})\times (\frac{i}{k}{)}^2 \\ =\sum _{i=1}^{n}p(\lfloor \frac{n}{i}\rfloor {)}^2\times i^2\sum _{k|i}\mu (\frac{i}{k})\times k \\ =\sum _{i=1}^{n}p(\lfloor \frac{n}{i}\rfloor {)}^2\times i^2\sum _{k|i}\mu (k)\times \frac{i}{k} \end{gathered}$$

发现后面的这个 $\sum _{k|i}\mu (k)\times \frac{i}{k}$ 就是 $\mu \ast id=\phi$ 的形式，所以有
$$=\sum _{i=1}^{n}p(\lfloor \frac{n}{i}\rfloor {)}^2\times i^2\times \phi (i)$$

后面的 $i^2\times \phi (i)$ 是可以用杜教筛搞的，于是就做完了。

然后再推一下如何用杜教筛来搞这个东西。

设 $f(n)=n^2\times \phi (n),f\ast g=h$，那么有
$$\begin{gathered} (f\ast g)(n)=\sum _{d|n}f(d)\times g(\frac{n}{d}) \\ (f\ast g)(n)=\sum _{d|n}\phi (d)\times i^2\times g(\frac{n}{d}) \end{gathered}$$

发现这个 $i^2$ 很不和谐，于是考虑 $g(n)=id(n{)}^2$，那么有
$$\begin{gathered} (f\ast g)(n)=\sum _{d|n}\phi (d)\times d^2\times id(\frac{n}{d}{)}^2 \\ (f\ast g)(n)=\sum _{d|n}\phi (d)\times d^2\times (\frac{n}{d}{)}^2 \\ (f\ast g)(n)=n^2\sum _{d|n}\phi (d) \\ (f\ast g)(n)=n^3 \end{gathered}$$

于是 $f,g,h$ 都定下来了，而且 $h$ 很好求，正是我们想要的，于是继续推柿子。
$$\begin{gathered} \sum _{i=1}^{n}h(i)=\sum _{i=1}^n\sum _{d|i}f(d)\times g(\frac{i}{d}) \\ \sum _{i=1}^{n}h(i)=\sum _{i=1}^{n}g(i)\sum _{j=1}^{\lfloor \frac{n}{i}\rfloor }f(j) \\ \sum _{i=1}^{n}h(i)=\sum _{i=1}^{n}g(i)S(\lfloor \frac{n}{i}\rfloor ) \\ \sum _{i=1}^{n}h(i)=S(n)+\sum _{i=2}^{n}g(i)S(\lfloor \frac{n}{i}\rfloor ) \\ S(n)=\sum _{i=1}^{n}h(i)-\sum _{i=2}^{n}g(i)S(\lfloor \frac{n}{i}\rfloor ) \end{gathered}$$

两个小性质：

$1^3+2^3+3^3+...+n^3=(\frac{n(n+1)}{2}{)}^2$

$1^2+2^2+3^2+...+n^2=\frac{1}{6}n(n+1)(2n+1)$（这个在代码里面用到，用来处理下面的 $i^2$）

于是有
$$S(n)=p(n{)}^2-\sum _{i=2}^n{i}^{2}S(\lfloor \frac{n}{i}\rfloor )$$

注意，上面的除以 $2$ 和除以 $6$ 别忘了要用逆元。

以及……大数据的 $n$ 是大于 $p$ 的，以至于这题的取模相当讲究……

于是就做完了！代码如下：

#include <cstdio>
#include <cstring>
#include <map>
#include <algorithm>
using namespace std;
#define maxn 6000010
#define ll long long
#define p(x) (x%mod*((x+1ll)%mod)%mod*inv2%mod)
#define w(x) ((x)%mod*inv6%mod*((x)%mod+1ll)%mod*(2ll*((x)%mod)%mod+1ll)%mod)

ll n,mod,inv2,inv6;
int prime[maxn],t=0;
ll phi[maxn];
bool v[maxn];
ll ans=0;
ll ksm(ll x,int y)
{
	ll re=1,tot=x;
	while(y>0)
	{
		if(y%2==1)re=re*tot%mod;
		y/=2;
		tot=tot*tot%mod;
	}
	return re;
}
void work()
{
    phi[1]=1;//先预处理phi的前6000000位，优化杜教筛复杂度
    for(int i=2;i<=maxn-10;i++)
    {
        if(!v[i])prime[++t]=i,phi[i]=i-1;
        for(int j=1;j<=t&&i*prime[j]<=maxn-10;j++)
        {
            v[i*prime[j]]=true;
            if(i%prime[j]==0)
            {
                phi[i*prime[j]]=phi[i]*prime[j];
                break;
            }
            phi[i*prime[j]]=phi[i]*phi[prime[j]];
        }
    }
    for(int i=2;i<=maxn;i++)
    phi[i]=phi[i]*(ll)i%mod*(ll)i%mod,phi[i]=(phi[i]+phi[i-1])%mod;
    //让phi[i]乘上i^2，就得到了f(i)，再求个前缀和，就变成了S(i)
    inv2=ksm(2ll,mod-2);//预处理2和6的逆元
    inv6=ksm(6ll,mod-2);
}
map<ll,ll>mp;
ll solve(ll x)
{
	if(x<=maxn-10)return phi[x];
	if(mp[x])return mp[x];
	ll ans=p(x)*p(x)%mod;
	ll l=2,r;
	while(l<=x)
	{
		r=x/(x/l);
		ans=(ans-solve(x/l)*(w(r)-w(l-1)+mod)%mod+mod)%mod;
		//柿子都跟上面的一样，没啥好说的
		l=r+1;
	}
	mp[x]=ans;
	return ans;
}

int main()
{
    scanf("%lld %lld",&mod,&n);
    work();
    ll l=1,r;
    while(l<=n)
    {
    	r=n/(n/l);
    	ans=(ans+p(n/l)*p(n/l)%mod*(solve(r)-solve(l-1)+mod)%mod)%mod;
    	l=r+1ll;
    }
    printf("%lld\n",ans);
}