BZOJ4407: 於神之怒增強版

# BZOJ4407: 於神之怒增強版

題目描述

傳送門

題目分析

題目讓求:
\[ Ans=\sum_{i=1}^{n}\sum_{j=1}^{m}gcd(i,j)^k \]

能夠發現是一個比較正常的式子，咱們直接開始化：

\[ \begin{aligned} Ans&=\sum_{i=1}^{n}\sum_{j=1}^{m}gcd(i,j)^k\\ &=\sum_{d=1}^{min(n,m)}\sum_{i=1}^{n}\sum_{j=1}^{m}[gcd(i,j)=d]d^k\\ &=\sum_{d=1}^{min(n,m)}d^k\sum_{i=1}^{n}\sum_{j=1}^{m}[gcd(i,j)=d]\\ &=\sum_{d=1}^{min(n,m)}d^k\sum_{d\mid T}\mu(\frac{T}{d})\lfloor\frac{n}{T}\rfloor\lfloor\frac{m}{T}\rfloor\\ &=\sum_{T=1}^{min(n,m)}\lfloor\frac{n}{T}\rfloor\lfloor\frac{m}{T}\rfloor\sum_{d\mid T}d^k\mu(\frac{T}{d}) \end{aligned} \]
這個式子明顯前面能夠整除分塊。

能夠發現後面的是個積性函數，又因爲全部數據的\(k\)都是同樣的，因此能夠線篩出後面的部分。而後直接求解就能夠了。

#include <bits/stdc++.h>
using namespace std;
const int MAXN=5e6+7;
const int mo=1e9+7;
#define ll long long
int prime[MAXN],n,m;
bool vis[MAXN];
int phi[MAXN],f[MAXN],g[MAXN],ans,k;
inline int power(int x,int k)
{
    int cnt=1;
    while(k){
        if(k&1) cnt=1ll*cnt*x%mo;
        x=(1ll*x*x)%mo;k>>=1;
    }
    return cnt%mo;
}
inline void get(int N)
{
    f[1]=1;
    for(int i=2;i<=N;i++){
        if(!vis[i]){
            prime[++prime[0]]=i;
            g[prime[0]]=power(i,k)%mo;
            f[i]=1ll*(1ll*g[prime[0]]-1+mo)%mo;
        }
        for(int j=1;j<=prime[0];j++){
            if(i*prime[j]>N) break;
            vis[i*prime[j]]=1;
            if(i%prime[j]==0){
                f[i*prime[j]]=1ll*f[i]*g[j]%mo;
                break;
            } else f[i*prime[j]]=1ll*f[i]*f[prime[j]]%mo;
        }
    }
    for(int i=1;i<=N;i++) f[i]=1ll*(1ll*f[i]+f[i-1])%mo;
}
inline int read()
{
    int x=0,c=1;
    char ch=' ';
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    while(ch=='-')c*=-1,ch=getchar();
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return x*c;
}
int main()
{
    int T=read();k=read();get(5000000);
    while(T--){
        n=read(),m=read();
        int mx=min(n,m);ans=0;
        for(int l=1,r;l<=mx;l=r+1){
            r=min(n/(n/l),m/(m/l));
            ans=(ans+1ll*(n/l)*(m/l)%mo*1ll*(1ll*f[r]-f[l-1]+mo)%mo)%mo;
        }
        printf("%d\n", ans);
    }
}