P4450 雙親數

題目:

​​題目連結:​​

題解:

莫比烏斯反演

即求

=

=

#include <bits/stdc++.h>
using namespace std;
const int N=1e6+10;
int mu[N],st[N],pri[N],con,sum[N];
void getmus()
{
    mu[1]=1;
    for(int i=2;i<=N;i++)
    {
        if(!st[i])
        {
            pri[++con]=i;
            mu[i]=-1;
        }
        for(int j=1;i*pri[j]<=N;j++)
        {
            int t=i*pri[j];
            st[t]=1;
            if(i%pri[j]==0)
            {
                break;
            }
            mu[t]=-mu[i];
        }
    }
    for(int i=1;i<=N;i++) sum[i]=sum[i-1]+mu[i];
}
long long f(int n,int m,int k)
{
    n/=k,m/=k;
    int x=min(n,m);
    long long ans=0;
    for(int i=1,j=1;i<=x;i=j+1)
    {
        j = min(x,min(n/(n/i),m/(m/i)));
        ans += (sum[j] - sum[i-1]) *(long long)(n/i) * (m/i);
    }
    return ans;
}
int main()
{
    getmus();
    ios::sync_with_stdio(false);
    cin.tie(0);cout.tie(0);
    {
        int a,b,k;
        cin>>a>>b>>k;
        long long ans=f(a,b,k);
        cout<<ans<<endl;
    }
    return 0;
}