CF Unbearable Controversy of Being 489D 暴力

题意:n点m条边的有向图,问有多少个四元组(a,b,c,d)满足存在边(a,b),(b,c),(a,d),(d,c).
n<=3e3,m<=3e4.


设f[a][b] 为a->b有多少条长度为2的路径
枚举一个点a,在枚举它的边(a,d),在枚举d的边(d,b)即可.
时间复杂度为O(2*m+n^2) 每条边都被枚举两次.
f[a][b]有x条长度为2的路径  则(a,b)有C(x,2)个四元组

也可以枚举边(a,b) 然后在O(n)枚举c 判断g[b][c]是否为真即可.O(n*m).

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e3+20,inf=0x3f3f3f3f;
int n,m,f[N][N];
vector<int> e[N];
int main()
{
	cin>>n>>m;
	int u,v;
	for(int i=0;i<m;i++)
	{
		scanf("%d%d",&u,&v);
		e[u].push_back(v);
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=0;j<e[i].size();j++)
		{
			int v=e[i][j];
			for(int k=0;k<e[v].size();k++)
			{
				int c=e[v][k];
				f[i][c]++;
			}
		}
	}
	int ans=0;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			if(i!=j)
				ans+=f[i][j]*(f[i][j]-1)/2;
		}
	cout<<ans<<endl;
	return 0;
	
}