Codeforces Round 942 (Div. 2) D2. Reverse Card (Hard Version) 题解 数论

D2. Reverse Card (Hard Version)

题目描述

给你两个正整数 $n$ ， $m$ 。

请计算满足以下条件的有序数对 $(a,b)$ 的个数：

• $1\le a\le n$ , $1\le b\le m$ ;
• $b\cdot \gcd (a,b)$ 是 $a+b$ 的倍数。

输入描述

每个测试包含多个测试用例。第一行包含测试用例的数量 $t$ ( $1\le t\le 10^4$ )。测试用例说明如下。

每个测试用例的第一行包含两个整数 $n$ , $m$ ( $1\le n,m\le 2\cdot 10^6$ )。

保证所有测试用例中 $n$ 和 $m$ 的总和不超过 $2\cdot 10^6$ 。

输出描述

为每个测试用例打印一个整数：有效配对的数量。

样例输入

6
1 1
2 3
3 5
10 8
100 1233
1000000 1145141

样例输出

0
1
1
6
423
5933961

原题

CF——传送门

思路

CF题解——传送门
令 $\gcd (a,b)=d$ ，则 $a=pd$ 和 $b=qd$ ，所以 $\gcd (p,q)=1$ 。于是条件可等价推导：$(a+b)\mid (b\cdot \gcd (a,b))⟺(pd+qd)\mid (q{d}^2)⟺(p+q)\mid (qd)$ 。又因为 $\gcd (p+q,q)=\gcd (p,q)=1$ ，所以$(p+q)\nmid q$，所以 $(p+q)\mid d$ 。此外，因为 $p\ge 1,q\ge 1$ ， $p<d=\frac{a}{p}\le \frac{n}{p}$ ，所以 $p^2<n$ 。同理， $q^2<m$ 。所以 $(p,q)$ 的个数是 $\mathcal{O}(\sqrt{nm})=\mathcal{O}(n+m)$ 。所以可以枚举 $\gcd (p,q)=1$ 的每个 $(p,q)$ ，找出其中满足 $(p+q)\mid d$ 的序对，即加上 $\lfloor \frac{\min \{\lfloor \frac{n}{p}\rfloor ,\lfloor \frac{m}{q}\rfloor \}}{p+q}\rfloor$ 。

PS：这场Div.2的C题被Hack了，由于二分查找上界写小了。所以我很难过。 所以以后要注意仔细想一下边界，不要想当然了（）。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    int t;
    cin >> t;
    while (t--)
    {
        ll n, m;
        cin >> n >> m;

        ll maxp = sqrt(n) + 6;
        ll maxq = sqrt(m) + 6;
        vector<vector<bool>> not_gcd_equal_to_1(maxp + 1, vector<bool>(maxq + 1, 0));
        // 预处理所有gcd(p,q)≠1的序对
        for (ll k = 2; k <= min(maxp, maxq); k++)
        {
            for (ll i = k; i <= maxp; i += k)
            {
                for (ll j = k; j <= maxq; j += k)
                {
                    not_gcd_equal_to_1[i][j] = 1;
                }
            }
        }
        ll ans = 0;
        for (ll p = 1; p * p <= n; p++)
        {
            for (ll q = 1; q * q <= m; q++)
            {
                if (!not_gcd_equal_to_1[p][q])
                {
                    ans += min(n / p / (p + q), m / q / (p + q));
                }
            }
        }
        cout << ans << '\n';
    }

    return 0;
}