题目大意
给定
n
,
m
,
k
n,m,k
n,m,k,求
∑
i
=
1
n
∑
j
=
1
m
[
(
i
,
j
)
=
k
]
\sum\limits_{i=1}^n\sum\limits_{j=1}^m[(i,j)=k]
i=1∑nj=1∑m[(i,j)=k]
题解
考虑化简,得:
∑
i
=
1
⌊
n
k
⌋
∑
j
=
1
⌊
m
k
⌋
[
(
i
,
j
)
=
1
]
\sum\limits_{i=1}^{\lfloor \frac nk\rfloor}\sum\limits_{j=1}^{\lfloor \frac mk\rfloor}[(i,j)=1]
i=1∑⌊kn⌋j=1∑⌊km⌋[(i,j)=1]
利用中心结论,得:
∑
i
=
1
⌊
n
k
⌋
∑
j
=
1
⌊
m
k
⌋
∑
d
∣
(
i
,
j
)
μ
(
d
)
\sum\limits_{i=1}^{\lfloor \frac nk\rfloor}\sum\limits_{j=1}^{\lfloor \frac mk\rfloor}\sum\limits_{d|(i,j)}\mu(d)
i=1∑⌊kn⌋j=1∑⌊km⌋d∣(i,j)∑μ(d)
∑ i = 1 ⌊ n k ⌋ ∑ j = 1 ⌊ m k ⌋ ∑ d = 1 ( i , j ) μ ( d ) [ d ∣ ( i , j ) ] \sum\limits_{i=1}^{\lfloor \frac nk\rfloor}\sum\limits_{j=1}^{\lfloor \frac mk\rfloor}\sum\limits_{d=1}^{(i,j)}\mu(d)[d|(i,j)] i=1∑⌊kn⌋j=1∑⌊km⌋d=1∑(i,j)μ(d)[d∣(i,j)]
∑ i = 1 ⌊ n k ⌋ ∑ j = 1 ⌊ m k ⌋ ∑ d = 1 ( i , j ) μ ( d ) [ d ∣ i ] [ d ∣ j ] \sum\limits_{i=1}^{\lfloor \frac nk\rfloor}\sum\limits_{j=1}^{\lfloor \frac mk\rfloor}\sum\limits_{d=1}^{(i,j)}\mu(d)[d|i][d|j] i=1∑⌊kn⌋j=1∑⌊km⌋d=1∑(i,j)μ(d)[d∣i][d∣j]
∵
g
c
d
(
a
,
b
)
⩽
m
i
n
(
a
,
b
)
\because gcd(a,b) \leqslant min(a,b)
∵gcd(a,b)⩽min(a,b)
设答案为
∑ d = 1 m i n ( ⌊ n k ⌋ , ⌊ m k ⌋ ) p d ∗ μ ( d ) 。 \sum\limits_{d=1}^{min(\lfloor \frac nk\rfloor,\lfloor \frac mk\rfloor)} p_d*\mu(d)。 d=1∑min(⌊kn⌋,⌊km⌋)pd∗μ(d)。
我们发现,
p
d
p_d
pd 等于
d
∣
i
d|i
d∣i 且
d
∣
j
d|j
d∣j 的数的个数。
因此可以转换成如下形式:
∑
d
=
1
m
i
n
(
⌊
n
k
⌋
,
⌊
m
k
⌋
)
μ
(
d
)
∑
i
=
1
⌊
n
k
⌋
[
d
∣
i
]
∑
j
=
1
⌊
m
k
⌋
[
d
∣
j
]
\sum\limits_{d=1}^{min(\lfloor \frac nk\rfloor,\lfloor \frac mk\rfloor)}\mu(d)\sum\limits_{i=1}^{\lfloor \frac nk\rfloor}[d|i]\sum\limits_{j=1}^{\lfloor \frac mk\rfloor}[d|j]
d=1∑min(⌊kn⌋,⌊km⌋)μ(d)i=1∑⌊kn⌋[d∣i]j=1∑⌊km⌋[d∣j]
后面两个和式可以直接得出:
∑
d
=
1
m
i
n
(
⌊
n
k
⌋
,
⌊
m
k
⌋
)
μ
(
d
)
⌊
n
k
d
⌋
⌊
m
k
d
⌋
\sum\limits_{d=1}^{min(\lfloor \frac nk\rfloor,\lfloor \frac mk\rfloor)}\mu(d)\lfloor\dfrac n{kd}\rfloor\lfloor\dfrac m{kd}\rfloor
d=1∑min(⌊kn⌋,⌊km⌋)μ(d)⌊kdn⌋⌊kdm⌋
线筛出
μ
\mu
μ,求前缀和,最后数论分块即可。时间复杂度
Θ
(
n
+
T
n
)
\Theta(n+T\sqrt{n})
Θ(n+Tn)。
#include<bits/stdc++.h>
using namespace std;
#define maxn 100000
int mu[maxn],p[maxn];
bool vis[maxn];
void euler_mu(int n) {
int tot=0;
memset(mu,0,sizeof mu);
memset(vis,0,sizeof vis);
mu[1]=1;
for(int i=2;i<=n;++i) {
if(!vis[i]) p[++tot]=i,mu[i]=-1;
for(int j=1;j<=tot&&i*p[j]<=n;++j) {
vis[i*p[j]]=1;
if(i%p[j]==0) {
mu[i*p[j]]=0;
break;
}
mu[i*p[j]]=-mu[i];
}
} for(int i=1;i<=n;i++)//前缀和
mu[i]+=mu[i-1];
}
int main(){
int T;
euler_mu(50000);//线性筛只做一次
scanf("%d",&T);
while(T--){
int _n,_m,k,ans=0;
scanf("%d%d%d",&_n,&_m,&k);
_n/=k;_m/=k;
int a=_n,b=_m,n=min(_n,_m);
int len=min(n,min(a,b));
for(int l=1,r;l<=len;l=r+1){
r=min(a/(a/l),b/(b/l));
if(r>n) r=n;
ans+=(mu[r]-mu[l-1])*(a/l)*(b/l);
}
printf("%d\n",ans);
}
return 0;
}