【bzoj4176】Lucas的数论 莫比乌斯反演+杜教筛

时间：2022-07-22 07:21:20

Description

去年的Lucas非常喜欢数论题，但是一年以后的Lucas却不那么喜欢了。

在整理以前的试题时，发现了这样一道题目“求Sigma(f(i)),其中1<=i<=N”，其中 表示i的约数个数。他现在长大了，题目也变难了。

求如下表达式的值：

【bzoj4176】Lucas的数论 莫比乌斯反演+杜教筛

其中 表示ij的约数个数。

他发现答案有点大，只需要输出模1000000007的值。

Input

第一行一个整数n。

Output

一行一个整数ans，表示答案模1000000007的值。

Sample Input

2

Sample Output

8

HINT

对于100%的数据n <= 10^9。

Sol

这个题的最大难点，在于对\(f(ij)\)的变形，只要这个变形正确了，后面的就是更换求和指标经典套路+数论分块经典套路了。

\(f(ij)=\sum_{x|i}\sum_{y|j}[(x,y)=1]\)

证明：ij的某个因子一定是i的某个因子*j的某个因子乘起来的，我们不妨设为i和\(\frac{j}{y}\)的某个因子，那么设\(p=(x,y)\)，那么你在x中包括了p这个因子，又在\(\frac{j}{y}\)中把它消掉，就没意义了，也就会重复统计，所以只有\((x,y)=1\)的时候才会有合法的贡献。

然后有了\((x,y)=1\)这个条件，直接上莫比乌斯函数：

\(\sum^{n}_{i=1}\sum_{j=1}^{n}f(i,j)=\sum^{n}_{i=1}\sum_{j=1}^{n}\sum_{x|i}\sum_{y|j}\sum_{d|x,d|y}\mu(d)\)

后面一步就是更换求和指标啦，把d提到最前面，x,y其次，i,j最后面，因为有两部分完全相同，所以这个式子就变成了：

$\sum\limits_{d=1}^n \mu(d)(\sum\limits_{i=1}^{\lfloor {n\over i}\rfloor} {\lfloor {n\over i*d}\rfloor})^2 $

先对于n/i分块，然后对于n/(i/d)分块，前面的莫比乌斯函数根据n/i的分块范围在线使用杜教筛计算。

时间复杂度$O(n{3/4}+n{2/3}logn) $

Code

#include <bits/stdc++.h>

#define ll long long

using namespace std;

map<ll,ll>mmp;int sum[1000005],pri[1000005],vis[1000005],tot,mu[1000005],n,ls,ans,P=1000000007;

ll djs(int x)

{

    if(x<=1e6) return sum[x];

    if(mmp.count(x)) return mmp[x];

    int ans=1,ls;

    for(int i=2;i<=x;i=ls+1) ls=x/(x/i),ans=(ans-1ll*(ls-i+1)*djs(x/i)%P+P)%P;

    return mmp[x]=ans;

}

ll cal(ll x)

{

    ll ans=0,ls;

    for(int i=1;i<=x;i=ls+1) ls=x/(x/i),ans=(ans+1ll*(ls-i+1)*(x/i)%P)%P;

    return 1ll*ans*ans%P;

}

int main()

{

    mu[1]=sum[1]=1;

    for(int i=2;i<=1000000;i++)

    {

        if(!vis[i]) pri[++tot]=i,mu[i]=-1;

        for(int j=1;j<=tot&&i*pri[j]<=1000000;j++)

        {

            vis[i*pri[j]]=1;

            if(i%pri[j]==0){mu[i*pri[j]]=0;break;}

            mu[i*pri[j]]=-mu[i];

        }

        sum[i]=(sum[i-1]+mu[i]+P)%P;

    }

    scanf("%d",&n);

    for(int i=1;i<=n;i=ls+1) ls=n/(n/i),ans=(ans+1ll*(djs(ls)-djs(i-1)+P)%P*cal(n/i)%P)%P;

    printf("%d\n",ans);

}

相关文章