题意
n (≤3000)个点
m (≤30000)条边的有向图,统计下图这种菱形的数量。
分析
一开始以为可以超过四个点构成菱形(如果这样怎么做?)。
两点之间有 num 条通过一个点连接的路,那么就有 num(num−1)2 个菱形。
直接暴力枚举每个点可构成的菱形数量,代码与官方题解略有不同,不是直接枚举两点。
代码
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<cctype>
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<iomanip>
#include<sstream>
#include<limits>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const ll INF = 1e18;
const int maxn = 2e5+10;
const ll MOD = 1000000007;
const double EPS = 1e-10;
const double Pi = acos(-1.0);
vector<int> G[maxn];
int num[maxn];
int main(){
#ifdef LOCAL
//freopen("C:\\Users\\lanjiaming\\Desktop\\acm\\in.txt","r",stdin);
//freopen("output.txt","w",stdout);
#endif
//ios_base::sync_with_stdio(0);
int n,m;
cin>>n>>m;
for(int i = 0; i < m; i++)
{
int u,v;
cin>>u>>v;
G[u].push_back(v);
}
int ans = 0;
for(int i = 1; i <= n; i++)
{
memset(num,0,sizeof num);
for(int j = 0; j < G[i].size(); j++)
{
int u = G[i][j];
for(int k = 0; k < G[u].size(); k++)
if (G[u][k] != i) ans += num[G[u][k]],num[G[u][k]]++;
}
}
cout<<ans<<endl;
return 0;
}