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;
}