bzoj 3309: DZY Loves Math 莫比乌斯反演

Description

对于正整数n，定义 $f(n)$ 为 $n$ 所含质因子的最大幂指数。例如
$f(1960)=f(2^3\ast 5^1\ast 7^2)=3,f(10007)=1,f(1)=0$。
给定正整数a,b，求${\sum }_{i=1}^n{\sum }_{j=1}^{m}f(gcd(i,j))$。

Input

第一行一个数T，表示询问数。
接下来T行，每行两个数a,b，表示一个询问。

Output

对于每一个询问，输出一行一个非负整数作为回答。

Sample Input

4

7558588 9653114

6514903 4451211

7425644 1189442

6335198 4957

Sample Output

35793453939901

14225956593420

4332838845846

15400094813
HINT

【数据规模】

T<=10000

1<=a,b<=10^7

分析：
推一波式子可得，
$$\sum _{i=1}^n\sum _{j=1}^{m}f(gcd(i,j))$$
$$\sum _{D=1}^{n}f(D)\sum _{i=1}^{\lfloor \frac{n}{D}\rfloor }\mu (d)\ast \lfloor \frac{n}{Dd}\rfloor \ast \lfloor \frac{m}{Dd}\rfloor$$
$$\sum _{T=1}^n\lfloor \frac{n}{T}\rfloor \lfloor \frac{m}{T}\rfloor \sum _{d|T}\mu (d)\ast f(\frac{T}{d})$$
现在就是求后面那个东西的前缀和。
设$g(T)={\sum }_{d|T}\mu (d)\ast f(\frac{T}{d})$，
因为要想$\mu (d)$不为$0$，满足
$$d=p_1^{{\alpha }_1}\ast p_2^{{\alpha }_2}\ast ....\ast p_k^{{\alpha }_k}$$
其中${\alpha }_i=[0,1]$。
考虑对于一个$T$，他的幂次只有减$1$或不变。也就是说，一开始最大的数，改变后还是最大的。
我们把${\alpha }_i$分成两类，$A$为最大的数（包括和他一样的），$B$为不是最大的数。
组合数知识可以知道，在$B$中选出奇数个和偶数个的方案数是一样的，除非$B$为空集。
假设$B$不为空，此时无论$A$如何选，$B$中奇偶数一样，而贡献正好相反，所以总贡献为$0$。
如果$B$为空，说明$A$全是同一个数，除非全选最大值才会变。如果不变恰好为$0$，所以减了$1$贡献为
$-(-1{)}^k$，$k$为质数个数。
考虑怎样求这个东西。我们先找到所有无平方因子数，然后枚举他的幂，枚举幂其实复杂度是很小的，可以看做线性。

代码：

/**************************************************************
    Problem: 3309
    User: liangzihao
    Language: C++
    Result: Accepted
    Time:17588 ms
    Memory:313792 kb
****************************************************************/
 
#include <iostream>
#include <cstdio>
#include <cmath>
#define LL long long
 
const int maxn=1e7+7;
 
using namespace std;
 
LL T,a,b,n,m,ans,cnt;
LL prime[maxn],not_prime[maxn],mu[maxn],f[maxn];
 
void get(LL n)
{
    mu[1]=1;
    for (LL i=2;i<=n;i++)
    {
        if (!not_prime[i])
        {
            prime[++cnt]=i;
            mu[i]=-1;
        }
        for (LL j=1;j<=cnt;j++)
        {
            if (prime[j]*i>n) break;
            not_prime[i*prime[j]]=1;
            if (i%prime[j]==0)
            {
                mu[i*prime[j]]=0;
                break;
            }
            mu[i*prime[j]]=-mu[i];
        }
    }    
    f[1]=0;
    for (LL i=2;i<=n;i++)
    {
        if (mu[i])
        {
            for (LL j=i;j<=n;j=j*i)
            {
                f[j]=-mu[i];
            }
        }
    }           
    for (LL i=1;i<=n;i++) f[i]+=f[i-1];
}
 
int main()
{
    scanf("%lld",&T);
    get(1e7);   
    while (T--)
    {
        scanf("%lld%lld",&n,&m);        
        if (n>m) swap(n,m);
        ans=0;
        for (LL i=1,last;i<=n;i=last+1)
        {
            last=min(n/(n/i),m/(m/i));
            ans+=(n/i)*(m/i)*(f[last]-f[i-1]);
        }
        printf("%lld\n",ans);
    }
}