BZOJ2154 || 洛谷P1829 [国家集训队]Crash的数字表格【莫比乌斯反演】

Time Limit: 20 Sec
Memory Limit: 259 MB

Description

今天的数学课上，Crash小朋友学习了最小公倍数(Least Common Multiple)。对于两个正整数a和b，LCM(a, b)表示能同时被a和b整除的最小正整数。例如，LCM(6, 8) = 24。回到家后，Crash还在想着课上学的东西，为了研究最小公倍数，他画了一张NM的表格。每个格子里写了一个数字，其中第i行第j列的那个格子里写着数为LCM(i, j)。一个45的表格如下： 1 2 3 4 5 2 2 6 4 10 3 6 3 12 15 4 4 12 4 20 看着这个表格，Crash想到了很多可以思考的问题。不过他最想解决的问题却是一个十分简单的问题：这个表格中所有数的和是多少。当N和M很大时，Crash就束手无策了，因此他找到了聪明的你用程序帮他解决这个问题。由于最终结果可能会很大，Crash只想知道表格里所有数的和mod 20101009的值

Input

输入的第一行包含两个正整数，分别表示N和M

Output

输出一个正整数，表示表格中所有数的和mod 20101009的值

Hint

100%的数据满足N, M ≤ 10^7。

---

题目分析

首先题目要求的式子为
$$\sum _{i=1}^n\sum _{j=1}^m\frac{i\ast j}{gcd(i,j)}$$

尝试把gcd提出来枚举

$$\sum _{d=1}^{min(n,m)}\sum _{i=1}^n\sum _{j=1}^m\frac{i\ast j}{d}[gcd(i,j)==d]$$

考虑枚举$gcd(x,y)=1$，则一定有$gcd(x\ast d,y\ast d)=d$

$$\sum _{d=1}^{min(n,m)}\sum _{x=1}^{\lfloor \frac{n}{d}\rfloor }\sum _{y=1}^{\lfloor \frac{m}{d}\rfloor }\frac{xd\ast yd}{d}[gcd(x,y)==1]=\sum _{d=1}^{min(n,m)}d\sum _{x=1}^{\lfloor \frac{n}{d}\rfloor }\sum _{y=1}^{\lfloor \frac{m}{d}\rfloor }x\ast y[gcd(x,y)==1]$$

回忆莫比乌斯函数的性质${\sum }_{d|n}\mu (d)=[n=1]$并带入得

$$\sum _{d=1}^{min(n,m)}d\sum _{x=1}^{\lfloor \frac{n}{d}\rfloor }\sum _{y=1}^{\lfloor \frac{m}{d}\rfloor }x\ast y\sum _{z|gcd(x,y)}\mu (z)=\sum _{d=1}^{min(n,m)}d\sum _{x=1}^{\lfloor \frac{n}{d}\rfloor }\sum _{y=1}^{\lfloor \frac{m}{d}\rfloor }x\ast y\sum _{z=1}^{gcd(x,y)}\mu (z)[z|gcd(x,y)]$$

接下来把$z$得枚举提前
$$\sum _{d=1}^{min(n,m)}d\sum _{z=1}^{min(\lfloor \frac{n}{d}\rfloor ,\lfloor \frac{m}{d}\rfloor )}\mu (z)\sum _{x=1}^{\lfloor \frac{n}{d}\rfloor }\sum _{y=1}^{\lfloor \frac{m}{d}\rfloor }x\ast y\ast [z|gcd(x,y)]$$

若有$z|gcd(x,y)$，则一定有$z|x,z|y$
所以考虑把枚举$x,y$变成枚举$z$的倍数，最后把$z^2$提前

$$\sum _{d=1}^{min(n,m)}d\sum _{z=1}^{min(\lfloor \frac{n}{d}\rfloor ,\lfloor \frac{m}{d}\rfloor )}\mu (z)\sum _{t_1=1}^{\lfloor \frac{n}{dz}\rfloor }\sum _{t_2=1}^{\lfloor \frac{m}{dz}\rfloor }{z}^2{t}_1{t}_2=\sum _{d=1}^{min(n,m)}d\sum _{z=1}^{min(\lfloor \frac{n}{d}\rfloor ,\lfloor \frac{m}{d}\rfloor )}\mu (z)\ast z^2\sum _{t_1=1}^{\lfloor \frac{n}{dz}\rfloor }{t}_1\sum _{t_2=1}^{\lfloor \frac{m}{dz}\rfloor }{t}_2$$

到这里可以发现$\mu (z)\ast z^2$可以$O(n)$线性筛预处理前缀和，最后那一坨就是等差数列求和
${\sum }_{z=1}^{min(\lfloor \frac{n}{d}\rfloor ,\lfloor \frac{m}{d}\rfloor )}$和${\sum }_{t_1=1}^{\lfloor \frac{n}{dz}\rfloor }{t}_1{\sum }_{t_2=1}^{\lfloor \frac{m}{dz}\rfloor }{t}_2$显然可以整除分块处理
渐进复杂度约为$O(n)$

(P.S. 注意减法取膜的问题)

---

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long lt;
#define Sum(x) ((1ll*x*(x+1)/2)%mod)

int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return f*x;
}

const int mod=20101009;
const int maxn=1e7+10;
int miu[maxn];
int vis[maxn],prim[maxn],cnt;
lt sum[maxn];

void Miu(int n)
{
    miu[1]=1;
    for(int i=2;i<=n;++i)
    {
        if(!vis[i]){ prim[++cnt]=i; miu[i]=-1;}
        for(int j=1;j<=cnt;++j)
        {
            if(i*prim[j]>n) break;
            vis[i*prim[j]]=1;
            if(i%prim[j]==0) break;
            else miu[i*prim[j]]=-miu[i];
        }
    }
    for(int i=1;i<=n;++i) 
    sum[i]=(sum[i-1]+1ll*i*i%mod*(miu[i]+mod)%mod)%mod;
}

lt qsum(int n,int m) 
{
    lt res=0; int lim=min(n,m);
    for(int ll=1,rr;ll<=lim;ll=rr+1) 
    {
        rr=min(n/(n/ll),m/(m/ll));
        res+=1ll*((sum[rr]-sum[ll-1])%mod+mod)%mod * Sum(n/ll)%mod * Sum(m/ll)%mod;
        res%=mod;
    }
    return res;
}

lt query(int n,int m) 
{
    lt res=0; int lim=min(n,m);
    for(int ll=1,rr;ll<=lim;ll=rr+1) 
    {
        rr=min(n/(n/ll),m/(m/ll));
        res+=1ll*(rr-ll+1)*(ll+rr)/2%mod * qsum(n/ll,m/ll)%mod;
        res%=mod;
    }
    return res;
}

int main() 
{
    int n=read(),m=read(); 
    Miu(min(n,m));
    printf("%lld\n",query(n,m));
    return 0;
}