[BZOJ2693]jzptab（莫比乌斯反演）

题目描述

传送门

题解

刚开始感觉这题就是道水题啊，又一看模数tm不是质数啊
同样假设$n<m$

$$\sum\limits_{i=1}^n\sum\limits_{j=1}^m[i,j]$$
$$=\sum\limits_{i=1}^n\sum\limits_{j=1}^m{ij\over (i,j)}$$
$$=\sum\limits_{i=1}^n\sum\limits_{j=1}^m\sum\limits_{d=1}^n[(i,j)=d]{ij\over d}$$
$$=\sum\limits_{d=1}^n\sum\limits_{i=1}^{n\over d}i\sum\limits_{j=1}^{m\over d}j[(i,j)=1]d$$
利用反演公式 $[n=1]=\sum\limits_{d|n}\mu(d)$
$$=\sum\limits_{d=1}^n\sum\limits_{t=1}^{n\over d}\sum\limits_{i=1}^{n\over d}i[t|i]\sum\limits_{j=1}^{m\over d}j[t|j]\mu(t)d$$
可以发现 $\sum\limits_{i=1}^{n\over d}i[t|i]$实际上就是将 $[1..{n\over d}]$中t的所有倍数都加起来，那么令 $s(n)=\sum\limits_{i=1}^n i={n(n+1)\over 2}$，那么实际上 $\sum\limits_{i=1}^{n\over d}i[t|i]=s({n\over dt})t$
于是原式
$$=\sum\limits_{d=1}^n\sum\limits_{t=1}^{n\over d}s({n\over dt})s({m\over dt})t^2\mu(t)d$$
令 $T=dt$，那么现在 $d$为$T$的约数
$$=\sum\limits_{T=1}^ns({n\over T})s({m\over T})\sum\limits_{d|T}({T\over d})^2\mu({T\over d})d$$
令 $F(n)=\sum\limits_{d|n}({n\over d})^2\mu({n\over d})d$， $f(n)=n^2\mu(n)$， $g(n)=n$
这就是一个狄利克雷卷积的形式， $F=f\times g$
可以发现 $f$和$g$都是完全积性函数，那么 $F$也是个积性函数
所以$F$可以根据莫比乌斯函数的性质用线性筛筛出来
显然当p为质数时， $F(n)=-p^2+p$，当 $(a,b)=1$时， $F(a)=F(a)F(b)$
那么需要注意的就是 $(a,b)\ne 1$的情况，也就是如何用 $F(a)$和p求出 $F(ap)$
显然对于 $a=\prod\limits_ip_i^{k_i}$的某一个质因子 $p^k$来说，只有 $\mu(1)$和 $\mu(p)$是有价值的
$F(p^k)=f(1)p^k+f(p)p^{k-1}$， $F(p^{k+1})=f(1)p^{k+1}+f(p)p^k$
也就是说每一项会多一个p
那么 $F(ap)=F(a)*p$
这样就能求出来 $F(n)$了
回代原式
$$=\sum\limits_{T=1}^ns({n\over T})s({m\over T})F(T)$$
只有 $O(\sqrt n+\sqrt m)$种取值
分块就可以了
时间复杂度 $O(n+T(\sqrt n+\sqrt m))$

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define LL long long
#define N 10000005
#define Mod 100000009

int T,n,m,ans;
int p[N],prime[N];LL f[N],s[N];

void get(int n)
{
    f[1]=1;s[1]=1;
    for (int i=2;i<=n;++i)
    {
        if (!p[i])
        {
            prime[++prime[0]]=i;
            f[i]=(i-(LL)i*i%Mod+Mod)%Mod;
        }
        s[i]=(LL)i*(i+1)/2%Mod;
        for (int j=1;j<=prime[0]&&i*prime[j]<=n;++j)
        {
            p[i*prime[j]]=1;
            if (i%prime[j]==0)
            {
                f[i*prime[j]]=f[i]*prime[j]%Mod;
                break;
            }
            else f[i*prime[j]]=f[i]*f[prime[j]]%Mod;
        }
    }
    for (int i=1;i<=n;++i) f[i]=(f[i]+f[i-1])%Mod;
}
int main()
{
    get(10000000);
    scanf("%d",&T);
    while (T--)
    {
        scanf("%d%d",&n,&m);
        if (n>m) swap(n,m);
        ans=0;
        for (int i=1,j=0;i<=n;i=j+1)
        {
            j=min(n/(n/i),m/(m/i));
            ans+=s[n/i]*s[m/i]%Mod*(f[j]-f[i-1])%Mod;
            ans=(ans%Mod+Mod)%Mod;
        }
        printf("%d\n",ans);
    }
}