《大白书》105页 单色三角形（组合计数）

时间限制：1秒  内存限制：64M

【问题描述】

　　给定空间里的n个点，其中没有3个点共线。每两个点之间都用红色或黑色线段连接。求3条边同色的三角形个数。

【输入格式】

　　第一行一个整数n（n<=1000），表示点数。
　　接下来的n行表示一个n*n的矩阵，第i行第j列的数字表示连接点i与点j的线段的颜色，1表示红色，2表色黑色。当然i=j时为0。输入中i行j列与j行i列一定形同。

【输出格式】

　一个整数表示答案。

【输入样例】

6
0 1 1 2 1 2
1 0 1 1 2 2
1 1 0 1 1 1
2 1 1 0 2 2
1 2 1 2 0 2
2 2 1 2 2 0

【输出样例】

6

【数据范围】

n<=1000

【来源】

《大白书》105页

暴力n^3很简单，但是不能过。

我们可以逆向思考一下求不是同色的三角形数，然后用总三角形减取这个数就可以了。

总三角形数就是C(n,3)，不同色的我们可以枚举每个点，将每个点的红色和黑色的个数相乘就可以得到过这个点，有一红一黑的三角形的个数。

但由于每个不同色三角形按上述算法会算2遍，所有还要除以2.

代码如下：

#include<cstdlib>
#include<cstdio>
#include<iostream>
using namespace std;
const int maxn=1005;

int a[maxn],b[maxn];
int n;

int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    scanf("%d",&n);
    int id;
    int ans=n*(n-1)*(n-2),ans2=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            scanf("%d",&id);
            if(id==1) a[i]++;
            if(id==2) b[i]++;
        }
        ans2+=a[i]*b[i];
    }
    cout<<ans/6-ans2/2;
    return 0;
}