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