算法集训队第二场考核赛_F.数组重构

原题链接：算法集训队第二场考核赛_F.数组重构

F.数组重构

题目描述

给定一个由 $n$ 个整数组成的数组 $a$ 。
如果 $1\le i<j\le n$ 且 $gcd(a_i,2a_j)>1$（其中 $gcd(x,y)$ 是 $x$ 和 $y$ 的最大公约数），我们称一对索引 $i,j$ 为良好。
如果您可以以任意方式对数组 $a$ 重新排序，请找到最大数量的良好索引对。

输入描述

第一行包含一个整数 $t(1\le t\le 1000)$——测试用例的数量。
测试用例的第一行包含一个整数 $n(2\le n\le 2000)$——数组中元素的数量。
测试用例的第二行包含 $n$ 个整数 $a_1,a_2,\dots ,a_n(1\le a_i\le 10^5)$ 。
保证所有测试用例的 $n$ 总和不超过 $2000$ 。

输出描述

对于每个测试用例，输出一个整数——良好索引对的最大数量。

输入输出样例

输入 #1：

3
4
3 6 5 3
2
1 7
5
1 4 2 4 1

输出 #1：

4
0
9

说明/提示：

【样例解释】：
在第一个例子中，数组元素可以重新排列如下：$[6,3,5,3]$ 。
在第三个例子中，数组元素可以重新排列如下：$[4,4,2,1,1]$ 。

解决办法：

如果 $a_i$ 的值是偶数，那么 $gcd(a_i,2a_j)$ 至少为2，而不需要考虑 $a_j$ 的值。

因此，我们可以将所有的偶数值放在奇数值之前（顺序不影响结果）。

此时剩下的就是如何排列奇数值部分了。事实上，它们的顺序并不重要，因为 $gcd(a_i,2a_j)=gcd(a_i,a_j)(对于奇数值部分的a_i和a_j)$​

这意味着无论奇数元素的顺序如何，每一对都将被视为 $1$ 次。​

参考代码：

#include <bits/stdc++.h>

using namespace std;

int main() {
    int t;
    cin >> t;
    while (t--) {
        int n;
        cin >> n;
        vector<int> a(n);
        for (int& x : a) cin >> x;
        sort(a.begin(), a.end(), [](int x, int y) {
            return x % 2 < y % 2;
            });
        int ans = 0;
        for (int i = 0; i < n; ++i) {
            for (int j = i + 1; j < n; ++j) {
                ans += __gcd(a[i], a[j] * 2) > 1;
            }
        }
        cout << ans << endl;
    }
}