BZOJ 2916 Monochromatic Triangles

Posted on 2017-01-17 12:13  ziliuziliu  阅读(148)  评论(0编辑  收藏  举报

bitset暴力。

正解其实也很简单啊。。。。就是数不同色三角形的个数,也就是每个点引出的异色边的对数/2。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<bitset>
#define maxn 1005
using namespace std;
int n,m,ans=0,x,y;
bitset <maxn> map[maxn],regis;
int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y);
        map[x][y]=1;map[y][x]=1;
    }    
    for (int i=1;i<=n;i++)
        for (int j=i+1;j<=n;j++)
        {
            if (!map[i][j]) continue;
            regis=map[i]&map[j];
            ans+=regis.count();    
        }
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        {
            if (i==j) continue;
            map[i][j]=map[i][j]^1;
        }
    for (int i=1;i<=n;i++)
        for (int j=i+1;j<=n;j++)
        {
            if (!map[i][j]) continue;
            regis=map[i]&map[j];
            ans+=regis.count();    
        }
    printf("%d\n",ans/3);
    return 0;
}