spoj 4491 PGCD - Primes in GCD Table （莫比乌斯反演）

题目链接：哆啦A梦传送门

题意：给出n，m，求${\sum }_{i=1}^n{\sum }_{j=1}^m[gcd(i,j)为素数]$
$n,m<=1e7$

依题意得：
$\begin{array}{rl}sum& =\sum _{p\in 素数}\sum _{i=1}^n\sum _{j=1}^{m}gcd(i,j)=p\end{array}$

这里我们用莫比乌斯反演第二描述去做。

$F(n)={\sum }_{n|d}f(d),f(n)={\sum }_{n|d}u(\frac{d}{n})F(d)$


设 f(i)表示满足$gcd(x,y)=i$时(x,y)的对数。
则F(i)表示满足$i|gcd(x,y)$时(x,y)的对数。我们易得：$F(i)=\lfloor \frac{n}{i}\rfloor \lfloor \frac{m}{i}\rfloor$

故$f(i)={\sum }_{i|d}u(\frac{d}{i})\lfloor \frac{n}{d}\rfloor \lfloor \frac{m}{d}\rfloor$

即：我们直接枚举$\frac{d}{i}$的倍数d，得：
$\begin{array}{r}sum=\sum _{p\in 素数}\sum _{d=1}^{\lfloor \frac{n}{p}\rfloor }u(d)\lfloor \frac{n}{pd}\rfloor \lfloor \frac{m}{pd}\rfloor \end{array}$

这个式子复杂度为$O(质数个数\ast \sqrt{n})$，复杂度还是有点大。

我们再令：
$$k=pd$$
那么代入得：
$\begin{array}{r}sum=\sum _{k=1}^n\sum _{p\in 素数,p|k}u(\frac{k}{p})\lfloor \frac{n}{k}\rfloor \lfloor \frac{m}{k}\rfloor \end{array}$

设 $\begin{array}{r}f(x)=\sum _{p\in 素数,p|x}u(\frac{x}{p})\end{array}$，那么
最终 $\begin{array}{r}sum=\sum _{k=1}^{n}f(k)\lfloor \frac{n}{k}\rfloor \lfloor \frac{m}{k}\rfloor \end{array}$

我们只需在线性筛那里把$f(x)$的前缀和弄出来就可以了。

线性筛方程：
$\begin{array}{r}f(x)=\sum _{p\in 素数,p|x}u(\frac{x}{p}),设x的最小质因子为y，x=i\ast y\end{array}$

$\begin{array}{rlr}& u(1),& x为素数\\ f(x)=& u(i),& i\%y==0\\ & -f(i)+u(i),& i\%y!=0\end{array}$
详情见：神犇

那么我们就能在$O(\sqrt{n})$的时间复杂度解决。

代码：

#include<bits/stdc++.h>

using namespace std;
typedef long long LL;

#define INF 0x3f3f3f3f
const LL mod=1e7+9;
const int N=1e7+10;

int prime[N],tot,mu[N];
bool vis[N];

LL f[N];///预处理函数f的前缀和


void init()
{
    mu[1]=1;

    for(int i=2;i<N;i++)
    {
        if(!vis[i]){
            prime[++tot]=i;
            mu[i]=-1;
            f[i]=1;
        }
        for(int j=1;j<=tot&&i*prime[j]<N;j++){
            vis[i*prime[j]]=1;
            if(i%prime[j]==0){

                f[i*prime[j]]=mu[i];
                mu[i*prime[j]]=0;
                break;
            }
            mu[i*prime[j]]=-mu[i];
            f[i*prime[j]]=-f[i]+mu[i];


        }

        f[i]+=f[i-1];
    }

}

LL solve(int n,int m)
{

    LL re=0;

    int r;
    for(int l=1;l<=n;l=r+1){
        r=min(n/(n/l),m/(m/l));
        re=re+1LL*(f[r]-f[l-1])*(n/l)*(m/l);

    }
    return re;
}
int main()
{
    init();

    int ncase;
    scanf("%d",&ncase);

    LL n,m;
    while(ncase--)
    {
        scanf("%d%d",&n,&m);
        if(n>m) swap(n,m);

        printf("%lld\n",solve(n,m));
    }
    return 0;
}

预处理前缀和，还有另外一种方法：
参考博客：(https://www.cnblogs.com/zyb993963526/p/7253495.html)

#include<bits/stdc++.h>

using namespace std;
typedef long long LL;

#define INF 0x3f3f3f3f
const LL mod=1e7+9;
const int N=1e7+10;

int prime[N],tot,mu[N];
bool vis[N];

LL sum[N];///预处理函数f的前缀和


void init()
{
    mu[1]=1;

    for(int i=2;i<N;i++)
    {
        if(!vis[i]){
            prime[++tot]=i;
            mu[i]=-1;
        }
        for(int j=1;j<=tot&&i*prime[j]<N;j++){
            vis[i*prime[j]]=1;
            if(i%prime[j]==0){

                mu[i*prime[j]]=0;
                break;
            }
            mu[i*prime[j]]=-mu[i];
        }
    }

    ///预处理函数f的前缀和
    sum[0]=0;
    for(int i=1;i<=tot;i++)
    {
        for(int j=prime[i];j<N;j+=prime[i]){
            sum[j]+=mu[j/prime[i]];
        }
    }
    for(int i=1;i<N;i++){
        sum[i]+=sum[i-1];
    }


}

LL solve(int n,int m)
{

    LL re=0;

    int r;
    for(int l=1;l<=n;l=r+1){
        r=min(n/(n/l),m/(m/l));
        re=re+1LL*(sum[r]-sum[l-1])*(n/l)*(m/l);

    }
    return re;
}
int main()
{
    init();

    int ncase;
    scanf("%d",&ncase);

    LL n,m;
    while(ncase--)
    {
        scanf("%d%d",&n,&m);
        if(n>m) swap(n,m);

        printf("%lld\n",solve(n,m));
    }
    return 0;
}

我的标签：做个有情怀的程序员。