BZOJ 2045: 双亲数（莫比乌斯反演）

题目描述

传送门

题目大意：0 < <script type="math/tex" id="MathJax-Element-7"><</script>a<=A，0 < <script type="math/tex" id="MathJax-Element-8"><</script>b<=B，问有多少有序数对(a, b)满足gcd(a, b)=d。A，B<=10^6。

---

题解

这是一道适合底层数论选手切的数论“扫盲”题。

题目要求（假设$a<=b$）

$$\sum_{i=1}^a\sum_{j=1}^b[(i, j)=k]$$

就是

$$\sum_{i=1}^a\sum_{j=1}^b[k|i][k|j][({i\over k},{j\over k})=1]$$

根据社会主义套路，枚举倍数，令$i={i\over k},j={j\over k}$（即改变$i,j$表示的意义），得到

$$\sum_{i=1}^{\lfloor {a\over k}\rfloor }\sum_{j=1}^{\lfloor {b\over k}\rfloor }[(i,j)=1]$$

将布尔框用$\mu$消去

$$\sum_{i=1}^{\lfloor {a\over k}\rfloor }\sum_{j=1}^{\lfloor {b\over k}\rfloor }\sum_{t|(i,j)}\mu(t)$$

先枚举$t$

$$\sum_{t=1}^{\lfloor {a\over k}\rfloor }\mu(t)\lfloor{\lfloor{a\over{k}}\rfloor \over t}\rfloor\lfloor{\lfloor{b\over{k}}\rfloor\over t}\rfloor$$

于是先预处理出$\mu$，然后线性算出答案。

---

代码

#include <bits/stdc++.h>
#define maxn 1000100
#define temp (i * prime[j])
#define LL long long

using namespace std;

int A, B, d, cnt;
LL ans;

int prime[maxn], miu[maxn];
bool vis[maxn];

void Da(){
    miu[1] = 1;
    for(int i = 2; i <= A; i++){
        if(!vis[i]){
            prime[++cnt] = i;
            miu[i] = -1;
        }

        for(int j = 1; j <= cnt && temp <= A; j++){
            vis[temp] = true;
            if(i % prime[j] == 0){
                miu[temp] = 0;
                break;
            }
            else  miu[temp] = -miu[i];
        }
    }
}

int main(){

    scanf("%d%d%d", &A, &B, &d);

    if(A > B)  swap(A, B);

    A /= d;  B /= d;

    Da();

    for(int i = 1; i <= A; i++)
        ans += 1LL * miu[i] * (A / i) * (B / i);

    printf("%lld\n", ans);

    return 0;
} 

---