hdu 6184 Counting Stars(三元环计数)

题目链接

题解链接

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+10;
int n,m,from[maxn],to[maxn];
vector<int>G[maxn],s[maxn];
int in[maxn],re[maxn],rev[maxn],val[maxn];
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=1;i<=n;i++)
        {
            re[i]=0;
            in[i]=0;
            G[i].clear();
            s[i].clear();
        }
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&from[i],&to[i]);
            in[from[i]]++;
            in[to[i]]++;
            val[i]=0;
        }
        for(int i=1;i<=m;i++)
        {
            int x=from[i],y=to[i];
            if(in[x]<in[y]) G[x].push_back(y),s[x].push_back(i);
            else if(in[x]>in[y]) G[y].push_back(x),s[y].push_back(i);
            else if(x<y) G[x].push_back(y),s[x].push_back(i);
            else G[y].push_back(x),s[y].push_back(i);
        }
        for(int i=1;i<=m;i++)
        {
            int x=from[i],y=to[i];
            for(int j=G[x].size()-1;j>=0;j--)
                re[G[x][j]]=i+1,rev[G[x][j]]=s[x][j];
            for(int j=G[y].size()-1;j>=0;j--)
            {
                int z=G[y][j];
                if(re[z]==i+1)
                {
                    val[i]++;
                    val[s[y][j]]++;
                    val[rev[z]]++;
                }
            }
        }
        long long ans=0;
        for(int i=1;i<=m;i++)
            ans+=1LL*val[i]*(val[i]-1)/2;
        printf("%lld\n",ans);
    }
    return 0;
}