bzoj3994: [SDOI2015]约数个数和莫比乌斯反演

最新推荐文章于 2020-06-25 16:56:08 发布

原创最新推荐文章于 2020-06-25 16:56:08 发布 · 280 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

数学相关专栏收录该内容

64 篇文章

订阅专栏

本文针对bzoj3994题目进行了解析，该题目要求计算给定范围内所有数的约数个数之和。通过数学推导，将问题转化为更易于解决的形式，并使用分块预处理的方法来降低时间复杂度。

bzoj3994: [SDOI2015]约数个数和

Description

设d(x)为x的约数个数，给定N、M，求 $\sum\limits_i^N\sum\limits_j^Md(ij)$

Input

输入文件包含多组测试数据。
第一行，一个整数T，表示测试数据的组数。
接下来的T行，每行两个整数N、M。

Output

T行，每行一个整数，表示你所求的答案。

Sample Input

2
7 4
5 6

Sample Output

110
121

HINT

1<=N, M<=50000
1<=T<=50000

分析

结论： $d(nm)=\sum\limits_{i|n}\sum\limits_{j|m}[gcd(i,j)=1]$
~~结论看不出来，证明还是可以水一下的~~
证明看这里：PoPoQQQ大爷的题解
于是

a n s = \sum i N \sum j M \sum u | i \sum v | j [g c d (u, v) = 1]

$ans=\sum_i^N\sum_j^M\sum_{u|i}\sum_{v|j}[gcd(u,v)=1]$

= \sum i N \sum j M ⌊ N i ⌋ ⌊ M j ⌋ [g c d (i, j) = 1]

$=\sum_i^N\sum_j^M\lfloor\frac{N}{i}\rfloor\lfloor\frac{M}{j}\rfloor[gcd(i,j)=1]$

= \sum i N \sum j M ⌊ N i ⌋ ⌊ M j ⌋ \sum d | n ⋀ d | n μ (d)

$=\sum_i^N\sum_j^M\lfloor\frac{N}{i}\rfloor\lfloor\frac{M}{j}\rfloor\sum_{d|n \bigwedge d|n}\mu(d)$
令

i=id,j=jd $i=id, j=jd$

= \sum d N d \sum i ⌊ N d ⌋ ⌊ N i d ⌋ \sum j ⌊ M d ⌋ ⌊ M j d ⌋

$=\sum_d^Nd\sum_i^{\lfloor\frac{N}{d}\rfloor}\lfloor\frac{N}{id}\rfloor\sum_j^{\lfloor\frac{M}{d}\rfloor}\lfloor\frac{M}{jd}\rfloor$
此时，我们令

f(n)=∑in⌊ni⌋ $f(n)=\sum\limits_i^{n}\lfloor\frac{n}{i}\rfloor$
则

a n s = \sum d N d \cdot f (⌊ N d ⌋) \cdot f (⌊ M d ⌋)

$ans=\sum_d^Nd\cdot f(\lfloor\frac{N}{d}\rfloor) \cdot f(\lfloor\frac{M}{d}\rfloor)$
对于

f(n) $f(n)$ 我们分块预处理，复杂度

O(nn−√) $O(n\sqrt n)$ 对于ans，我们再分块，复杂度

O(Tn−√) $O(T\sqrt n)$

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;
const int N = 50000;
int read() {
    char ch = getchar();int x = 0, f = 1;
    while(ch < '0' || ch > '9') {if(ch == '-') f = -1; ch = getchar();}
    while(ch >= '0' && ch <= '9') {x = x * 10 + ch - '0'; ch = getchar();}
    return x * f; 
}
int mu[N + 10], p[N + 10], tot, n, m;
long long f[N + 10];
bool vis[N + 10];

void mobius() {
    mu[1] = 1;
    for(int i = 2; i <= N; ++i) {
        if(!vis[i]) {mu[i] = -1; p[++tot] = i;}
        for(int j = 1;j <= tot && i * p[j] <= N; ++j) {
            vis[i * p[j]] = true;
            if(i % p[j]) mu[i * p[j]] = -mu[i];
            else break;
        }
    }
    for(int i = 1;i <= N; ++i) mu[i] += mu[i - 1];
}

long long get_f(int n) {
    long long ret = 0;
    for(int i = 1, pos; i <= n; i = pos + 1) {
        pos = n / (n / i);
        ret += (n / i) * (pos - i + 1);
    }
    return ret;
}

int main() {
    mobius();
    for(int i = 1;i <= N; ++i) f[i] = get_f(i);
    int T = read();
    while(T--) {
        n = read(); m = read(); if(n > m) swap(n, m);
        long long ans = 0;
        for(int i = 1, pos;i <= n; i = pos + 1) {
            pos = min(n / (n / i), m / (m / i));
            ans += f[n / i] * f[m / i] * (mu[pos] - mu[i - 1]);
        }
        printf("%lld\n", ans);
    }

    return 0;
}