UVA 1510 Neon Sign(计数)

UVA 1510 Neon Sign

题目链接

题意：给定一些点两两相连，已知每两点连接是红色还是蓝色，问同色三角形有多少个

思路：由于不同色三角形也有两边同色，直接考虑不好考虑，反过来考虑，先找出不同色三角形，对于每个点而言，找一个红边和一个蓝边就能构成不同色三角形，那么每个三角形被选了3次，其中一次是同色的不用考虑，所以最后答案除以2，然后在用总情况数C(n, 3) - sum即可

代码：

#include <cstdio>#include <cstring>const int N = 1005;int t, n, col, red[N], black[N];int main() {    scanf("%d", &t);    while (t--) {memset(red, 0, sizeof(red));memset(black, 0, sizeof(black));scanf("%d", &n);for (int i = 0; i < n - 1; i++) {    for (int j = 0; j < n - 1 - i; j++) {scanf("%d", &col);if (col) {    red[i]++;    red[i + j + 1]++;}else {    black[i]++;    black[i + j + 1]++;}    }}int sum = 0;for (int i = 0; i < n; i++)    sum += black[i] * red[i];sum /= 2;printf("%d\n", n * (n - 1) * (n - 2) / 6 - sum);    }    return 0;}