题目大意:给定n个点,任意三点不共线,并且两两点之间有一条线,给定线的颜色。问说有多少个三角形三边同色。
解题思路:对于每个点,记录该点黑色边的数量和红色边的数量,考虑以该点为顶点的三角形,从红色边中选一条,黑色边中选一条,组成的三角形一定是不满足的。因为一个不同色三角形会有两个点满则,所以考虑了两次。用总的个数减掉不同色的即可。
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int maxn = 1005;
ll n, r[maxn], l[maxn];
void init () {
scanf("%lld", &n);
int x;
memset(l, 0, sizeof(l));
memset(r, 0, sizeof(r));
for (int i = 0; i < n; i++) {
for (int j = i + 1; j < n; j++) {
scanf("%d", &x);
if (x) {
r[i]++;
r[j]++;
} else {
l[i]++;
l[j]++;
}
}
}
}
ll solve () {
ll sum = (ll)n * (n-1) * (n-2) / 6;
ll del = 0;
for (int i = 0; i < n; i++)
del += l[i] * r[i];
return sum - del / 2;
}
int main () {
int cas;
scanf("%d", &cas);
while (cas--) {
init();
printf("%lld\n", solve());
}
return 0;
}
uva 1510 - Neon Sign(计数),布布扣,bubuko.com
时间: 2024-10-07 04:25:19