CodeForces 489 D.Unbearable Controversy of Being（组合数学+bfs）

Description

给出一张$n$个点$m$条边的有向图，问其中平行四边形的数量，平行四边形指的是存在四个不同的节点$a,b,c,d$满足$a\rightarrow b,b\rightarrow c,a\rightarrow d,d\rightarrow c$，计数时不考虑$b,d$的相对顺序

Input

第一行两个整数$n,m$，之后$m$行每行输入两个整数$u,v$表示一条有向边$u\rightarrow v(1\le n\le 3000,0\le m\le 30000)$

Output

输出平行四边形个数

Sample Input

5 4
1 2
2 3
1 4
4 3

Sample Output

1

Solution

枚举$a$，把$a$的所有一步可达点入队，然后对队列中每个点$u$，给其可达点$v$的次数加一，表示$v$点作为$c$点可以有多少点作为$b$点或$d$点，从中选俩做$b,d$即可，时间复杂度$O(m)$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
#include<bitset>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=3005;
int n,m,num[maxn];
vector<int>g[maxn]; 
ll bfs(int s)
{
    queue<int>que;
    memset(num,0,sizeof(num));
    for(int i=0;i<g[s].size();i++)que.push(g[s][i]);
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        for(int i=0;i<g[u].size();i++)
            if(g[u][i]!=s)num[g[u][i]]++;
    }
    ll res=0;
    for(int i=1;i<=n;i++)res+=(ll)num[i]*(num[i]-1)/2;
    return res;
}
int main()
{
    scanf("%d%d",&n,&m);
    while(m--)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        g[u].push_back(v);
    }   
    ll ans=0;
    for(int i=1;i<=n;i++)ans+=bfs(i);
    printf("%I64d\n",ans);
    return 0;
}