P1989 無向圖三元環計數

P1989 無向圖三元環計數

傳送門

思路：三元環模板題。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+5,M=1e6+5,inf=0x3f3f3f3f,mod=1e9+7;
#define mst(a) memset(a,0,sizeof a)
#define lx x<<1
#define rx x<<1|1
#define reg register
#define PII pair<int,int>
#define fi first 
#define se second
vector<int>e[N];
int n,m,deg[N],vis[N];
PII a[N<<1];
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++){
		scanf("%d%d",&a[i].fi,&a[i].se);
		deg[a[i].fi]++,deg[a[i].se]++;
	}
	for(int i=1;i<=m;i++){
		int u=a[i].fi,v=a[i].se;
		if(deg[u]<deg[v]||deg[u]==deg[v]&&u>v) swap(u,v);
		e[u].push_back(v);
	}
	ll ans=0;
	for(int i=1;i<=n;i++){
		for(auto u:e[i]) vis[u]=i;
		for(auto u:e[i]) 
			for(auto v:e[u])
				if(vis[v]==i) ans++;
	} 
	printf("%lld\n",ans);
	return 0;
}