【Wannafly Daily】【XJTUOJ】12 好友推荐

#include<stdio.h>
#include<cstring>
#include<vector>
using namespace std;
typedef long long ll;
#define pb push_back
#define N 100005
#define M 200005

vector<int> G1
,G2
,G
;
int sum1
,sum2
,sum3[M],n,m,du
,e[M][2];

#define For(i,v) for (int _##i=0,i;_##i<v.size() && (i=v[_##i]);_##i++)

inline ll sum_4()
{
ll ans1=0,ans2=0;
for (int i=1;i<=n;i++)
{
For(j,G1[i]) For(k,G1[j]) sum1[k]++;
For(j,G2[i]) For(k,G1[j]) sum2[k]++;
For(j,G1[i]) For(k,G1[j])
{
ans1+=(ll)sum1[k]*(sum1[k]-1)/2;
ans1+=(ll)sum1[k]*sum2[k];
sum1[k]=0;
}
For(j,G2[i])For(k,G1[j])
{
if (k!=i) ans2+=(ll)sum2[k]*(sum2[k]-1)/2;
sum2[k]=0;
}
}
return ans1+ans2/2;
}
inline ll sum_3()
{
for (int i=1;i<=m;i++)
{
int u=e[i][0],v=e[i][1];
For(j,G1[u]) sum1[j]=G[u][_j];
For(j,G1[v]) if (sum1[j])
{
sum3[i]++;
sum3[sum1[j]]++;
sum3[G[v][_j]]++;
}
For(j,G1[u]) sum1[j]=0;
}
ll ans=0;
for (int i=1;i<=m;i++) ans+=(ll)sum3[i]*(sum3[i]-1)/2;
return ans;
}

int main()
{
while (~scanf("%d%d",&n,&m))
{
for (int i=1;i<=n;i++) G1[i].clear(),G2[i].clear(),G[i].clear();
memset(du+1,0,n*sizeof(int));
memset(sum3+1,0,m*sizeof(int));
for (int i=1,u,v;i<=m;i++)
{
scanf("%d%d",&u,&v);
e[i][0]=u,e[i][1]=v;
du[u]++,du[v]++;
}
for (int i=1;i<=m;i++)
{
int u=e[i][0],v=e[i][1];
if (du[v]<du[u] || (du[u]==du[v] && v<u)) swap(u,v);
G1[u].pb(v);G2[v].pb(u);
G[u].pb(i);
}
printf("%lld\n",2*sum_4()-sum_3());
}
}