[国家集训队]和与积

Problem

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

Solution

• 设 $d=gcd(a,b)，a=di，b=dj$，那么根据题意有：

• $$(di+dj)|d^{2}ij$$

• 即：

• $$(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\le 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)}\mu (k)$$

• 设 $i=xk，j=yk$，那么:

• $$ans=\sum _{k=1}^{\sqrt{n}}\mu (k)\sum _{x=1}^{\lfloor \frac{\sqrt{n}}{k}\rfloor }\sum _{y=1}^{x-1}\lfloor \frac{\lfloor \frac{n}{x{k}^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;
}