BZOJ 3994: [SDOI2015]约数个数和 [莫比乌斯反演 转化]

时间:2021-08-06 01:24:19

2015

题意:\(d(i)\)为i的约数个数,求\(\sum\limits_{i=1}^n \sum\limits_{j=1}^m d(ij)\)


\(ij\)都爆int了....

一开始想容斥一下用\(d(i)\)和\(d(j)\)算\(d(ij)\),发现不行...

然后翻题解看到了一步好神的转化:

\[d(nm) = \sum_{i\mid n} \sum_{j\mid m} [gcd(i,j)=1]
\]

晚上再补吧还是没拿草稿纸...



补:

\(Proof.\)

  • 首先注意约数个数 相同的算一个

    约数个数公式\(\prod (a_i+1)\)

    考虑一个质因子,\(p^x,p^y \rightarrow p^x p^y\)

    \(x+y+1\)对应了\(gcd(p^x, 1)...gcd(1, 1)...gcd(1,p^y)\)

    质因子相互独立,乘起来

然后愉♂悦的套路推♂倒

\[\begin{align*}
\sum_{i=1}^n \sum_{j=1}^m d(ij) &= \sum_{i=1}^n \sum_{j=1}^m \sum_{x\mid i} \sum_{y\mid j} [gcd(x,y)=1]\\
先枚举约数,交换i,j\ x,y\\
&=\sum_{i=1}^n \sum_{j=1}^m \sum_{d\mid i,d\mid j}\mu(d) \frac{n}{i} \frac{m}{i}\\
&=\sum_{d=1}^n \mu(d)\sum_{i=1}^\frac{n}{i} \sum_{j=1}^\frac{m}{i} \frac{n}{id}\frac{m}{jd}\\
&=\sum_{d=1}^n \mu(d) f(\frac{n}{id})f(\frac{m}{jd})\\
\end{align*}
\]

问题就是\(f(n)=\sum_{i=1}^n\frac{n}{i}\)怎么求了

可以n根n预处理...

更巧妙的做法是,发现\(f\)就是\(d\)的前缀和,因为\(\frac{n}{i}\)表示的就是\(1..n\)有几个i的倍数

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=5e4+5;
typedef long long ll;
#define pii pair<int, int>
#define MP make_pair
#define fir first
#define sec second
inline int read(){
char c=getchar();int x=0,f=1;
while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
return x*f;
} int n, m;
int notp[N], p[N], mu[N]; ll f[N]; pii lp[N];
void sieve(int n) {
mu[1] = 1; f[1] = 1;
for(int i=2; i<=n; i++) {
if(!notp[i]) p[++p[0]] = i, mu[i] = -1, f[i] = 2, lp[i] = MP(i, 1);
for(int j=1; j<=p[0] && i*p[j]<=n; j++) {
int t = i*p[j];
notp[t] = 1;
if(i%p[j] == 0) {
mu[t] = 0;
lp[t] = MP(p[j], lp[i].sec + 1);
f[t] = f[i] / (lp[i].sec + 1) * (lp[t].sec + 1);
break;
}
mu[t] = -mu[i];
lp[t] = MP(p[j], 1);
f[t] = f[i] * (lp[t].sec + 1);
}
}
for(int i=1; i<=n; i++) mu[i] += mu[i-1], f[i] += f[i-1];
}
ll cal(int n, int m) {
ll ans=0; int r;
for(int i=1; i<=n; i=r+1) {
r = min(n/(n/i), m/(m/i));
ans += (mu[r] - mu[i-1]) * f[n/i] * f[m/i];
}
return ans;
}
int main() {
//freopen("in","r",stdin);
sieve(N-1);
int T=read();
while(T--){
n=read(); m=read();
if(n>m) swap(n, m);
printf("%lld\n",cal(n, m));
}
}