STIPC-003_编程挑战系列赛第三场（柯柯的期末祝福） _B.数组重构

原题链接：STIPC-003_B.数组重构

B.数组重构

题目描述

给定一个由 $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,...,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$ ：

如果 $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$ 次，因此对于奇数值来说，只要排列在偶数后即可，内部顺序不需要考虑。

AC代码（C++）：

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
bool cmp(int x, int y){ return x % 2 < y % 2;}
int main() 
{
    int t;
    cin >> t;
    while (t--) {
        int n, a[2001];
        cin >> n;
        for (int i = 0; i < n; i++) cin >> a[i];
        sort(a, a + n, cmp);
        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;
    }
}