bzoj 4407: 于神之怒加强版莫比乌斯反演

最新推荐文章于 2020-03-30 16:06:39 发布

_beginend

最新推荐文章于 2020-03-30 16:06:39 发布

阅读量437

点赞数

分类专栏：莫比乌斯反演

本文链接：https://blog.csdn.net/qq_33229466/article/details/54630304

版权

莫比乌斯反演专栏收录该内容

24 篇文章 0 订阅

订阅专栏

题意

给出n,m,k求
这里写图片描述
多组数据
n,m<=5000000,T<=2000

分析

$设d=gcd(i,j)$

$那么ans=\sum_{d=1}^{n}d^k*f(d)$

$f(d)表示1<=i<=n,1<=j<=m中gcd(i,j)=d的(i,j)对数$

$反演后可得ans=\sum_{d=1}^nd^k\sum_{i=1}^{\lfloor\frac{n}{d}\rfloor}\mu(i)\lfloor\frac{n}{di}\rfloor\lfloor\frac{m}{di}\rfloor$

$设T=di则ans=\sum_{T=1}^n\lfloor\frac{n}{T}\rfloor\lfloor\frac{m}{t}\rfloor\sum_{d|T}d^k\mu(\frac{T}{d})$

$那么只要求出\sum_{d|T}d^k\mu(\frac{T}{d})的前缀和即可O(\sqrt{n})询问$

$设f(d)=d^k,g(d)=\mu(d),h(T)=\sum_{d|T}f(d)g(\frac{T}{d})$

显然h为f和g的狄利克雷卷积，因为f和g均为积性函数，那么h也为积性函数，那么就可以用线性筛来求出h

low[i]表示i的最小质因子的其对应指数的次方。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define N 5000005
#define ll long long
using namespace std;

const int MOD=1000000007;

int k,n,m,tot,prime[N],low[N],f[N];
bool not_prime[N];

int ksm(int x,int y)
{
    if (!y) return 1;
    if (y==1) return x;
    int w=ksm(x,y/2);
    w=(ll)w*w%MOD;
    if (y%2==1) w=(ll)w*x%MOD;
    return w;
}

void prework(int n)
{
    f[1]=1;
    for (int i=2;i<=n;i++)
    {
        if (!not_prime[i])
        {
            low[i]=prime[++tot]=i;
            f[i]=ksm(i,k)-1;
        }
        for (int j=1;j<=tot&&i*prime[j]<=n;j++)
        {
            not_prime[i*prime[j]]=1;
            if (i%prime[j]==0)
            {
                if (low[i]==i) f[i*prime[j]]=(ll)f[i]*(f[prime[j]]+1)%MOD;
                else f[i*prime[j]]=(ll)f[low[i]*prime[j]]*f[i/low[i]]%MOD;
                low[i*prime[j]]=low[i]*prime[j];
                break;
            }
            f[i*prime[j]]=(ll)f[i]*f[prime[j]]%MOD;
            low[i*prime[j]]=prime[j];
        }
    }
    for (int i=1;i<=n;i++)
        f[i]=(f[i-1]+f[i])%MOD;
}

int main()
{
    int t;
    scanf("%d%d",&t,&k);
    prework(5000000);
    while (t--)
    {
        scanf("%d%d",&n,&m);
        if (n>m) swap(n,m);
        int ans=0;
        for (int i=1,last;i<=n;i=last+1)
        {
            last=min(n/(n/i),m/(m/i));
            ans=(ans+(ll)(n/i)*(m/i)%MOD*(f[last]-f[i-1])%MOD)%MOD;
        }
        printf("%d\n",(ans+MOD)%MOD);
    }
    return 0;
}