http://acm.hdu.edu.cn/showproblem.php?pid=3908
题意:给定一个数组,判断数组中这样的三元组的个数:要么三个元素两两互素,要么三个元素两两都不互素。
ans=C(n,3) - 三个数中有一对互素的并且有一对不互素的个数sum
sum怎么算?
对于num[i],计算与之互素的元素的个数pnum,那么不互素的元素的个数就是n-1-pnum了,对于num[i]而言,不满足条件的三元组个数为pnum*(n-1-pnum)。
注意最后还要除2:比如(2, 3, 4)在处理2的时候算了一次,在处理4的时候也算了一次,但是在处理3的时候,肯定不会出现,因为对3而言,2与3互素,4也与3互素,所以总的来说每个三元组都多算了一次。
完整代码:
/*1015ms,260KB*/
#include<cstdio>
const int maxn = 805;
int a[maxn];
int gcd(int a, int b) {return b ? gcd(b, a % b) : a;}
int main()
{
int t;
scanf("%d", &t);
while (t --)
{
int n;
scanf("%d", &n);
for (int i = 0; i < n; i++) scanf("%d", &a[i]);
int ans = 0;
for (int i = 0; i < n; i++)
{
int num = 0;
for (int j = 0; j < n; j++)
if (i != j && gcd(a[i], a[j]) == 1) ++num;
ans += num * (n - num - 1);
}
printf("%d\n", n * (n - 1) * (n - 2) / 6 - ans / 2);
}
return 0;
}