于神之怒加强版 [Bzoj 4407]

题目地址请点击——

---

于神之怒加强版

---

Description

给定 $N$ , $M$ , $K$ .

---

Input

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

---

Output

如题

---

Sample Input

1 2
3 3

---

Sample Output

20

---

Hint

$1<=N,M,K<=5000000,1<=T<=2000$

---

Solution

ans=∑t=1n⌊mt⌋⌊nt⌋⋅∑p|tpk⋅μ(tp)

$$ans=\sum^n_{t=1}\lfloor\frac{m}{t}\rfloor\lfloor\frac{n}{t}\rfloor·\sum_{p|t}p^k·\mu(\frac{t}{p})$$
所以我们只需要求出 $f(t)=\sum_{p|t}p^k·\mu(\frac{t}{p})$ 的前缀和即可。
注意到 $p^k$、$\mu(p)$ 均为积性函数，所以设 $g(p)=p^k$，则 $g*\mu$ 也为积性函数。
因为 $g*\mu=\sum_{p|t}p^k·\mu(\frac{t}{p})=f(t)$，所以 $f(t)$ 为积性函数。

所以设 $t=\prod_{i=1}^sp_i^{q_i}$，则 $f(t)=\prod^s_{i=1}(p_i^{q_ik}-p_i^{q_ik-k})$，然后用线性筛乱搞就可以了。

---

Code

#include <iostream>
#include <cstdio>
#include <ctime> 
#include <cmath>

#define LL long long
#define Min(x,y) ((x)<(y)?(x):(y))
#define MAXN (5000000)
#define MOD (1000000007)

using namespace std;

bool no_prime[MAXN+10];
LL prime[MAXN+10];
LL sum[MAXN+10];
LL p[MAXN+10];
LL f[MAXN+10]; 
LL n,m,k,T,ans;

LL powerm(LL x,LL y){
    if(p[x]!=0)return p[x];
    if(y==0)return p[x]=1;
    if(y==1)return p[x]=x%MOD;
    LL tmp=powerm(x,y/2);
    if(y&1)return p[x]=tmp*tmp%MOD*(x%MOD)%MOD;
    return p[x]=tmp*tmp%MOD;
}

int main(){

    scanf("%lld%lld",&T,&k);

    f[1]=1;
    sum[1]=1;

    for(LL i=2;i<=MAXN;i++){
        if(!no_prime[i]){prime[++prime[0]]=i;f[i]=(powerm(i,k)-1);}
        for(LL j=1;prime[j]*i<=MAXN;j++){
            no_prime[prime[j]*i]=true;
            if(i%prime[j]==0){
                f[prime[j]*i]=f[i]*(f[prime[j]]+1)%MOD;
                break;
            }
            f[prime[j]*i]=f[i]*f[prime[j]]%MOD;
        }
        sum[i]=(sum[i-1]+f[i])%MOD;
    }

    while(T--){
        scanf("%lld%lld",&n,&m);
        LL ans=0;
        if(n>m)swap(n,m);
        for(LL i=1,it;i<=n;i=it+1){
            it=Min(m/(m/i),n/(n/i));
            ans=(ans+(m/i)*(n/i)%MOD*((sum[it]-sum[i-1])%MOD)%MOD)%MOD;
        }
        printf("%lld\n",(ans+MOD)%MOD);
    }
    return 0;
}