【bzoj4407】于神之怒加强版 莫比乌斯反演+线性筛

题目描述

给下N,M,K.求

输入

输入有多组数据，输入数据的第一行两个正整数T,K，代表有T组数据，K的意义如上所示，下面第二行到第T+1行，每行为两个正整数N,M，其意义如上式所示。

输出

如题

样例输入

1 2
3 3

样例输出

20

---

题解

莫比乌斯反演+线性筛

$sumlimits_{i=1}^nsumlimits_{j=1}^mgcd(i,j)^k\=sumlimits_{d=1}^{min(n,m)}d^ksumlimits_{i=1}^nsumlimits_{j=1}^m[gcd(i,j)=d]\=sumlimits_{d=1}^{min(n,m)}d^ksumlimits_{i=1}^{lfloorfrac nd floor}sumlimits_{j=1}^{lfloorfrac md floor}[gcd(i,j)=1]\=sumlimits_{d=1}^{min(n,m)}d^ksumlimits_{i=1}^{lfloorfrac nd floor}sumlimits_{j=1}^{lfloorfrac md floor}sumlimits_{t|gcd(i,j)}mu(t)\=sumlimits_{d=1}^{min(n,m)}d^ksumlimits_{i=1}^{lfloorfrac nd floor}sumlimits_{j=1}^{lfloorfrac md floor}sumlimits_{t|i&t|j}mu(t)\=sumlimits_{d=1}^{min(n,m)}d^ksumlimits_{t=1}^{lfloorfrac{min(n,m)}d floor}mu(t)lfloorfrac n{dt} floorlfloorfrac m{dt} floor$

继续令$D=dt$，可以得到：

$sumlimits_{d=1}^{min(n,m)}d^ksumlimits_{t=1}^{lfloorfrac{min(n,m)}d floor}mu(t)lfloorfrac n{dt} floorlfloorfrac m{dt} floor\=sumlimits_{D=1}^{min(n,m)}lfloorfrac nD floorlfloorfrac mD floorsumlimits_{d|D}d^kmu(t)\=sumlimits_{D=1}^{min(n,m)}lfloorfrac nD floorlfloorfrac mD floor f(D)\(f(D)=sumlimits_{d|D}d^kmu(t))$

此时可以选择通过枚举倍数来预处理$f$数组，有个小优化：枚举时先枚举$t$，如果$mu(t)=0$则不进行操作。这样预处理的时间复杂度为$O(nln n+nlog k)$，可以勉强通过本题。

然后我就被CQzhangyu大佬D了一顿= =

此时可以发现$f$是$id^k$与$mu$的狄利克雷卷积，两个积性函数的狄利克雷卷积也是积性函数。于是可以快筛$f$函数。

当$i$为质数时，显然 $f(i)=i^k-1$ 

考虑 $i*p$ （p是质数）的处理：

当 $imod p eq 0$ 时，显然$i$与$p$互质，就有$f(i*p)=f(i)*f(p)$

当 $imod p=0$ 时，对$f(i*p)$有影响的$t$一定是与对$f(i)$有影响的$t$中，因为其余的因数都至少包含$p^2$，$mu=0$。而此时$d$增加了$p$倍，故$f(i*p)=f(i)*p^k$。

综上可以线性筛出$f$函数，然后分块处理。时间复杂度为$O(n+frac{nlog k}{ln n}+Tsqrt n)$

#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 5000010
using namespace std;
typedef long long ll;
const int p = 5000000 , mod = 1000000007;
int prime[N] , tot;
ll d[N] , f[N] , sum[N];
bool np[N];
ll pow(ll x , ll y)
{
    ll ans = 1;
    while(y)
    {
        if(y & 1) ans = ans * x % mod;
        x = x * x % mod , y >>= 1;
    }
    return ans;
}
int main()
{
    int T , k , i , j , n , m;
    ll ans;
    scanf("%d%d" , &T , &k);
    f[1] = sum[1] = 1;
    for(i = 2 ; i <= p ; i ++ )
    {
        if(!np[i]) prime[++tot] = i , d[tot] = pow(i , k) , f[i] = (d[tot] - 1 + mod) % mod;
        for(j = 1 ; j <= tot && i * prime[j] <= p ; j ++ )
        {
            np[i * prime[j]] = 1;
            if(i % prime[j] == 0)
            {
                f[i * prime[j]] = f[i] * d[j] % mod;
                break;
            }
            else f[i * prime[j]] = f[i] * f[prime[j]] % mod;
        }
        sum[i] = (sum[i - 1] + f[i]) % mod;
    }
    while(T -- )
    {
        scanf("%d%d" , &n , &m) , ans = 0;
        for(i = 1 ; i <= n && i <= m ; i = j + 1)
            j = min(n / (n / i) , m / (m / i)) , ans = (ans + (ll)(n / i) * (m / i) % mod * (sum[j] - sum[i - 1] + mod)) % mod;
        printf("%lld
" , ans);
    }
    return 0;
}