莫比乌斯反演+二维前缀和——hdu4746二刷

第二次做这题，求前缀和的时候还是卡住了

fg函数的反演是可以直接用莫比乌斯基本代换式来代换的

#include<bits/stdc++.h>
using namespace std;
#define maxn 500005
#define ll long long 

bool vis[maxn];
int mu[maxn],prime[maxn],mm,num[maxn];
void init(){//顺便筛出每个数的质因子个数
    mu[]=;num[]=;
    for(int i=;i<maxn;i++){
        if(!vis[i]){
            mu[i]=-;
            prime[++mm]=i;
            num[i]=;
        }
        for(int j=;j<=mm;j++){
            if(prime[j]*i>=maxn)break;
            vis[i*prime[j]]=;
            if(i%prime[j]==){
                num[i*prime[j]]=num[i]+;
                mu[i*prime[j]]=;
                break;
            }
            else {
                mu[i*prime[j]]=-mu[i];
                num[i*prime[j]]=num[i]+;
            }
        }
    }
}

long long n,m,p;
int sum[maxn][];//前缀和，质因子个数，等式值
void prework(){
    for(int k=;k<maxn;k++)
        for(int j=;j*k<maxn;j++)
            sum[j*k][num[k]]+=mu[j];

    for(int i=;i<maxn;i++)
        for(int j=;j<=;j++)
            sum[i][j]+=sum[i][j-];
    for(int i=;i<maxn;i++)
        for(int j=;j<=;j++)
            sum[i][j]+=sum[i-][j];
}

int main(){
    init();
    prework();
    int t;cin>>t;
    while(t--){
        scanf("%lld%lld%lld",&n,&m,&p);
        if(n>m)swap(n,m);
        if(p>=){
            printf("%lld\n",n*m);
            continue;
        }

        long long ans=;
        for(int l=,r;l<=n;l=r+){
            r=min(n/(n/l),m/(m/l));
            ans+=(sum[r][p]-sum[l-][p])*(n/l)*(m/l);
        }
        cout<<ans<<'\n';
    }
}