显然O(n^3)的算法会超时。正确的解法是首先计算出与每个数互质的数的个数以及与每个数不互质的数的个数,分别记入数组co[]和nco[]。这样,假设一个三元组包含数字x(x,..,..)。那么包含x且不符合题意的三元组个数是co[x]*nco[x]。比如一组数字2,3,4,5,7,8。对于数字2来说,与2互质的数字个数是3,不互质的数字个数是2。所以不满足题意的三元组数目是2*3=6。(2,3,4),(2,3,8),(2,5,4),(2,5,8),(2,7,4),(2,7,8)。然而对于某个三元组来说,我们计算了两次,比如(2,3,4)在数字2计算了一次,在数字4也计算了一次。所以结果要除以2。最后只需要将总的三元组数目减去不符合题意的三元组数目即可。
#include <stdio.h> #include <stdlib.h> #include <math.h> #include <string.h> #include <memory.h> #include <string> #include <vector> #include <map> #include <set> #include <queue> #include <algorithm> #include <iostream> #include <sstream> using namespace std; int gcd(int a, int b) { while (b) { int r = a % b; a = b; b = r; } return a; } int d[805], co[805], nco[805]; int main() { int t, n, i, j, sum; for (scanf("%d", &t); t--; ) { scanf("%d", &n); for (i = 0; i < n; i++) scanf("%d", &d[i]); memset(co, 0, 4 * n); memset(nco, 0, 4 * n); for (i = 0; i < n; i++) { for (j = i + 1; j < n; j++) { if (gcd(d[i], d[j]) == 1) co[i]++, co[j]++; else nco[i]++, nco[j]++; } } sum = 0; for (i = 0; i < n; i++) sum += co[i] * nco[i]; printf("%d\n", n * (n - 1) * (n - 2) / 6 - sum / 2); } return 0; }