bzoj3994/洛谷P3327 约数个数和 莫比乌斯反演

1.题目链接

https://www.luogu.org/problemnew/show/P3327

2.题面

题目描述

设d(x)为x的约数个数，给定N、M，求 \sum^N_{i=1}\sum^M_{j=1}d(ij)∑i=1N​∑j=1M​d(ij)

输入输出格式

输入格式：

输入文件包含多组测试数据。第一行，一个整数T，表示测试数据的组数。接下来的T行，每行两个整数N、M。

输出格式：

T行，每行一个整数，表示你所求的答案。

输入输出样例

输入样例#1： 复制

2
7 4
5 6

输出样例#1： 复制

110
121

说明

1<=N, M<=50000

1<=T<=50000

 

3.题意思路

再吐槽一下公式真的难推。

4.代码

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<cmath>
using namespace std;
#define MAX 50000
int n,m;
bool zs[MAX+1000];
int pri[MAX+1000],tot,mu[MAX+1000],ys[MAX+1000],dd[MAX+1000];
int smu[MAX+1000],sd[MAX+1000];
void pre()
{
    zs[1]=true;mu[1]=ys[1]=1;
    for(int i=2;i<=MAX;++i)
    {
        if(!zs[i])pri[++tot]=i,mu[i]=-1,ys[i]=2,dd[i]=1;
        for(int j=1;j<=tot&&i*pri[j]<=MAX;++j)
        {
            zs[i*pri[j]]=true;
            if(i%pri[j]==0)
            {
                mu[i*pri[j]]=0;
                ys[i*pri[j]]=ys[i]/(dd[i]+1)*(dd[i]+2);
                dd[i*pri[j]]=dd[i]+1;
                break;
            }
            else mu[i*pri[j]]=-mu[i],ys[i*pri[j]]=ys[i]*2,dd[i*pri[j]]=1;
        }
    }
    for(int i=1;i<=MAX;++i)smu[i]=smu[i-1]+mu[i];
    for(int i=1;i<=MAX;++i)sd[i]=sd[i-1]+ys[i];
}
int main()
{
    pre();
    int T;
    cin>>T;
    while(T--)
    {
        cin>>m>>n;
        if(n>m)swap(n,m);
        long long ans=0;
        int i=1,j;
        while(i<=n)
        {
            j=min(n/(n/i),m/(m/i));
            ans+=1ll*(smu[j]-smu[i-1])*sd[n/i]*sd[m/i];
            i=j+1;
        }
        printf("%lld\n",ans);
    }
    return 0;
}