于神之怒加强版 (莫反)

Posted 2021-02-13 nldqy

题目链接： 点击这里

首先哦，令n<=m，我们来化简式子：
[ sum_{i=1}^nsum_{j=1}^mgcd(i,j)^k ]
设gcd(i,j)=d，则可以变成这样:
[ sum_{i=1}^nsum_{j=1}^md^k\\]
把d提到前面去，来枚举d：
[ sum_{d=1}^nd^ksum_{i=1}^{lfloor{frac{n}{d}} floor}sum_{j=1}^{lfloor{frac{m}{d}} floor}[gcd(i,j)=1] ]
然后就可以开始套了
[ sum_{d=1}^nd^ksum_{i=1}^{lfloor{frac{n}{d}} floor}sum_{j=1}^{lfloor{frac{m}{d}} floor}sum_{x|i,x|j}mu(x) ]
然后找到满足条件的x的个数，就可以和前面两个sigma说再见了
[ sum_{d=1}^nd^ksum_{x=1}^{lfloor{frac{n}{d}} floor}mu(x)lfloor{frac{n}{dx}} floorlfloor{frac{m}{dx}} floor设T=dx，再来枚举T\\sum_{d=1}^nd^ksum_{x=1}^{lfloor{frac{n}{d}} floor}mu(frac{T}{d})lfloor{frac{n}{T}} floorlfloor{frac{m}{T}} floor\\sum_{T=1}^nlfloor{frac{n}{T}} floorlfloor{frac{m}{T}} floorsum_{d|T}d^kmu(frac{T}{d}) ]
一般的题到这一步就能直接搞了，然而会T，所以我们来继续推：

注：以下受@hby大佬的启发，与本人博客差不多

设
[ f(x)=sum_{d|T}d^kmu(frac{T}{d})g(x)=x^k ]

看！(f(x))是不是(/mu)和(g)的卷积形式！所以(f)是个积性函数!

我们来对(f(x))进行讨论，首先我们将(x)分解质因数
[ x=Pi_{p_i^{a_i}},p_iinmathbb{P}f(x)=Pi_{f(p_i^{a_i})}f(p_i^{a^i})=sum_{d|p_i^{a_i}}d^kmu(frac{p_i^{a_i}}{d})因为p_i是个质数，所以f(p_i^{a^i})=sum_{i=0}^{a_i}{p_i}^kmu(p_i^{a_i-i})再考虑一下mu的性质，可以得出只有当i=a^i或i=a^i-1时，mu不为0，于是f(p_i^{a_i})=p_i^{a_ik}-p_i^{(a_i-1)k}于是对于f(xp)，若p|x，则只是多了几个指数而已，那么f(xp)=f(x)*p^k否则，就是添加了几个质因数，则f(xp)=f(x)*(p^k-1) ]

Code:

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=5e6+1;
const int pps=1e9+7;
int n,m,k,t,tot;
int pri[N],mu[N],vis[N],p[N],f[N];
int quick(int a,int b){
    int sum=1;
    while(b){
        if(b&1) sum*=a,sum%=pps;
        a=(a*a)%pps,b>>=1;
    }return sum%pps;
}
void prepare(){
    f[1]=1;
    for(int i=2;i<=N;i++){
        if(!vis[i]) pri[++tot]=i,p[tot]=quick(i,k),f[i]=p[tot]-1;
        for(int j=1;j<=tot&&pri[j]*i<=N;j++){
            vis[i*pri[j]]=1;
            if(!(i%pri[j])){
                f[i*pri[j]]=f[i]*p[j]%pps;
                break;
            }f[i*pri[j]]=f[i]*(p[j]-1)%pps;
        }
    }for(int i=1;i<=N;i++) f[i]=(f[i]+f[i-1])%pps;
}
int calc(int n,int m){
    int d=1,sum=0;
    while(d<=n&&d<=m){
        int pre=d;d=min(n/(n/d),m/(m/d));
        sum=(sum+(n/d)*(m/d)%pps*(f[d]-f[pre-1])%pps)%pps;
        ++d;
    }return (sum+pps)%pps;
}
int read(){
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();}
    while(isdigit(ch)){x=x*10+ch-48;ch=getchar();}
    return x*f;
}
signed main(){
    t=read(),k=read();
    prepare();
    while(t--){
        n=read(),m=read();
        printf("%lld
",calc(n,m));
    }
    return 0;
}