原题链接:算法集训队第二场考核赛_F.数组重构
F.数组重构
题目描述
给定一个由 n n n 个整数组成的数组 a a a 。
如果 1 ≤ i < j ≤ n 1≤i<j≤n 1≤i<j≤n 且 g c d ( a i , 2 a j ) > 1 gcd(a_i,2a_j)>1 gcd(ai,2aj)>1(其中 g c d ( x , y ) gcd(x,y) gcd(x,y) 是 x x x 和 y y y 的最大公约数),我们称一对索引 i , j i, j i,j 为良好。
如果您可以以任意方式对数组 a a a 重新排序,请找到最大数量的良好索引对。输入描述
第一行包含一个整数 t ( 1 ≤ t ≤ 1000 ) t (1≤t≤1000) t(1≤t≤1000)——测试用例的数量。
测试用例的第一行包含一个整数 n ( 2 ≤ n ≤ 2000 ) n (2≤n≤2000) n(2≤n≤2000)——数组中元素的数量。
测试用例的第二行包含 n n n 个整数 a 1 , a 2 , … , a n ( 1 ≤ a i ≤ 1 0 5 ) a_1,a_2,…,a_n (1≤a_i≤10^5) a1,a2,…,an(1≤ai≤105) 。
保证所有测试用例的 n n n 总和不超过 2000 2000 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 ] [6,3,5,3] [6,3,5,3] 。
在第三个例子中,数组元素可以重新排列如下: [ 4 , 4 , 2 , 1 , 1 ] [4,4,2,1,1] [4,4,2,1,1] 。
解决办法:
如果 a i a_i ai 的值是偶数,那么 g c d ( a i , 2 a j ) gcd(a_i,2a_j) gcd(ai,2aj) 至少为2,而不需要考虑 a j a_j aj 的值。
因此,我们可以将所有的偶数值放在奇数值之前(顺序不影响结果)。
此时剩下的就是如何排列奇数值部分了。事实上,它们的顺序并不重要,因为 g c d ( a i , 2 a j ) = g c d ( a i , a j ) ( 对 于 奇 数 值 部 分 的 a i 和 a j ) gcd(a_i,2a_j)=gcd(a_i,a_j)(对于奇数值部分的a_i和a_j) gcd(ai,2aj)=gcd(ai,aj)(对于奇数值部分的ai和aj)
这意味着无论奇数元素的顺序如何,每一对都将被视为 1 1 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;
}
}