CodeForces 489D Unbearable Controversy of Being

题意：

给出一个n个节点m条边的有向图，求如图所示的菱形的个数。

这四个节点必须直接相邻，菱形之间不区分节点b、d的个数。

 

分析：

我们枚举每个a和c，然后求出所有满足a邻接t且t邻接c的节点的个数记为r。

那么分别以a、c为左右端点的菱形的个数就是r的二元组合。

#include <cstdio>
#include <vector>

using namespace std;

const int maxn = 3000 + 10;
bool G[maxn][maxn];
vector<int> nxt[maxn];

int main()
{
    //freopen("in.txt", "r", stdin);
    int n, m;
    scanf("%d%d", &n, &m);
    for(int i = 0; i < m; ++i)
    {
        int a, b;
        scanf("%d%d", &a, &b);
        G[a][b] = true;
        nxt[a].push_back(b);
    }

    int ans = 0;
    for(int a = 1; a <= n; ++a)
        for(int c = 1; c <= n; ++c)
        {
            if(a != c)
            {
                int r = 0;
                for(int b = 0; b < nxt[a].size(); ++b)
                {
                    if(nxt[a][b] != a && nxt[a][b] != c && G[nxt[a][b]][c])
                        r++;
                }
                ans += r*(r-1)/2;
            }
        }

    printf("%d\n", ans);

    return 0;
}