题目大意
设 d ( x ) d(x) d(x)为 x x x的约数个数,给定 n , m n,m n,m,求
∑ i = 1 n ∑ j = 1 m d ( i j ) \sum\limits_{i=1}^n\sum\limits_{j=1}^md(ij) i=1∑nj=1∑md(ij)
有 T T T组数据。
1 ≤ T , n , m ≤ 50000 1\leq T,n,m\leq 50000 1≤T,n,m≤50000
题解
前置知识:
式子的证明
首先,我们要知道:
d ( i j ) = ∑ x ∣ i ∑ y ∣ j [ gcd ( x , y ) = 1 ] d(ij)=\sum\limits_{x|i}\sum\limits_{y|j}[\gcd(x,y)=1] d(ij)=x∣i∑y∣j∑[gcd(x,y)=1]
证明:
如果 i j ij ij的因子 k k k中有因子 p c p^c pc, i i i中有因子 p a p^a pa, j j j中有因子 p b p^b pb,其中 p p p为质数。那么我们规定
- 当 c ≤ a c\leq a c≤a时,这个因子 p p p都在 a a a中选择
- 当 c > a c>a c>a时,在 i i i中选 p a p^a pa,在 j j j中选 p c − a p^{c-a} pc−a
对于 i j ij ij的因子 k k k的每个因子都这样处理。
如果令 x , y x,y x,y互质,则
- 若 x , y x,y x,y中都没有因子 p p p,则
- 若 x x x中有因子 p p p且次数为 k k k,则相当于在 x x x中选了 p k p^k pk
- 若 y y y中有因子 p p p且次数为 k k k,则相当于在 x x x中选了 p a p^a pa,在 j j j中选了 p k p^k pk
i j ij ij的各个因子 k k k对应各个互质的 x , y x,y x,y,所以等式成立。
推式子
根据上面的式子,有
∑ i = 1 n ∑ j = 1 m d ( i j ) = ∑ i = 1 n ∑ j = 1 m ∑ x ∣ i ∑ y ∣ j [ gcd ( x , y ) = 1 ] \sum\limits_{i=1}^n\sum\limits_{j=1}^md(ij)=\sum\limits_{i=1}^n\sum\limits_{j=1}^m\sum\limits_{x|i}\sum\limits_{y|j}[\gcd(x,y)=1] i=1∑nj=1∑md(ij)=i=1∑nj=1∑mx∣i∑y∣j∑[gcd(x,y)=1]
由莫比乌斯函数的性质,原式可变为
∑ i = 1 n ∑ j = 1 m ∑ x ∣ i ∑ y ∣ j ∑ k ∣ i , k ∣ j μ ( k ) \sum\limits_{i=1}^n\sum\limits_{j=1}^m\sum\limits_{x|i}\sum\limits_{y|j}\sum\limits_{k|i,k|j}\mu(k) i=1∑nj=1∑mx∣i∑y∣j∑k∣i,k∣j∑μ(k)
先枚举 k k k,则转换为
∑ k = 1 min ( n , m ) μ ( k ) ∑ i = 1 ⌊ n k ⌋ ∑ j = 1 ⌊ m k ⌋ ⌊ n k i ⌋ ⌊ m k j ⌋ \sum\limits_{k=1}^{\min(n,m)}\mu(k)\sum\limits_{i=1}^{\lfloor\frac nk\rfloor}\sum\limits_{j=1}^{\lfloor\frac mk\rfloor}\lfloor\dfrac{n}{ki}\rfloor\lfloor\dfrac{m}{kj}\rfloor k=1∑min(n,m)μ(k)i=1∑⌊kn⌋j=1∑⌊km⌋⌊kin⌋⌊kjm⌋
令 f ( n ) = ∑ i = 1 n ⌊ n i ⌋ f(n)=\sum\limits_{i=1}^n\lfloor\dfrac ni\rfloor f(n)=i=1∑n⌊in⌋,则从 1 1 1到 n n n的 f ( i ) f(i) f(i)都是可以 O ( n ) O(n) O(n)预处理出来的。
用数论块即可 O ( n ) O(\sqrt n) O(n)求出答案。
总时间复杂度为 O ( n + T n ) O(n+T\sqrt n) O(n+Tn)。
code
#include<bits/stdc++.h>
using namespace std;
const int N=100000;
int z[N+5],p[N+5],mu[N+5];
long long ans=0,smu[N+5],d[N+5];
void init(){
mu[1]=1;
for(int i=2;i<=N;i++){
if(!z[i]){
p[++p[0]]=i;mu[i]=-1;
}
for(int j=1;j<=p[0]&&i*p[j]<=N;j++){
z[i*p[j]]=1;
if(i%p[j]==0) break;
mu[i*p[j]]=-mu[i];
}
}
for(int i=1;i<=N;i++){
smu[i]=smu[i-1]+mu[i];
for(int j=i;j<=N;j+=i) ++d[j];
d[i]=d[i-1]+d[i];
}
}
int main()
{
int t,n,m;
init();
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&m);
if(n>m) swap(n,m);
ans=0;
for(int l=1,r;l<=n;l=r+1){
r=min(n/(n/l),m/(m/l));
ans=ans+(smu[r]-smu[l-1])*d[n/l]*d[m/l];
}
printf("%lld\n",ans);
}
return 0;
}