[国家集训队]和与积（莫比乌斯反演）

Problem

• 给出 \(n\) ，统计满足以下条件的数对 \((a,b)\) 的个数：
  1.\(1≤a<b≤n\)
  2.\(a+b|ab\)

• \(n<2^{31}\)

  Solution

• 设 \(d=gcd(a,b)，a=di，b=dj\)，那么根据题意有：
• \[(di+dj)|d^2ij\]
• 即：
• \[(i+j)|ijd\]
• 又因为：
• \[gcd(i,j)=1\]
• 所以：
• \[ij\%(i+j)!=0\]
• 那么：
• \[(i+j)|d\]

• 那么问题转化为求满足以下条件的三元组 \((i,j,d)\) 的个数：
  1.\(gcd(i,j)=1\)
  2.\(i<j\)
  3.\(di,dj≤n\)
• \[ans=\sum_{i=1}^{\sqrt{n}}\sum_{j=1}^{i-1}\lfloor{\frac{\lfloor{\frac{n}{i}}\rfloor}{i+j}}\rfloor [gcd(i,j)=1]\]
• \[ans=\sum_{i=1}^{\sqrt{n}}\sum_{j=1}^{i-1}\lfloor{\frac{n}{i(i+j)}}\rfloor\sum_{k|gcd(i,j)}μ(k)\]
• 设 \(i=xk，j=yk\)，那么:
• \[ans=\sum_{k=1}^{\sqrt{n}}μ(k)\sum_{x=1}^{\lfloor{\frac{\sqrt{n}}{k}}\rfloor}\sum_{y=1}^{x-1}\lfloor{\frac{\lfloor{\frac{n}{xk^2}}\rfloor}{x+y}}\rfloor\]

• 那么枚举 \(x,k\)，然后对 \(x+y\) 整除分块即可。

Code

#include <bits/stdc++.h>

using namespace std;

#define ll long long

const int e = 1e6 + 5;
int n, mul[e];
bool bo[e];
ll ans;

inline ll solve(int x, int k)
{
    int m = 2 * x, j, t = n / (x * k * k), i;
    ll res = 0;
    for (i = x + 1; i < m; i = j + 1)
    {
        if (!(t / i)) return res;
        j = min(m - 1, t / (t / i));
        if (j - i + 1 >= 0) res += (ll)(j - i + 1) * (t / i);
    }
    return res;
}

int main()
{
    cin >> n;
    int s = sqrt(n), x, k, i, j;
    for (i = 1; i <= s; i++) mul[i] = 1;
    for (i = 2; i <= s; i++)
    if (!bo[i])
    {
        mul[i] = -1;
        for (j = 2 * i; j <= s; j += i)
        {
            bo[j] = 1;
            if (j / i % i == 0) mul[j] = 0;
            else mul[j] *= -1;
        }
    }
    for (k = 1; k <= s; k++)
    if (mul[k])
    for (x = 1; x * k * k <= n && x * k <= s; x++)
    ans += mul[k] * solve(x, k);
    cout << ans << endl;
    return 0;
}