求点对数

Problem Description

给定一个有向图，统计有多少点对u,v(1≤u<v≤n) 满足u 可以到达v ，且v 可以到达u 。

Input

第一行包含一个正整数T(1≤T≤10) ，表示测试数据的组数。

每组数据第一行包含两个正整数n,m(1≤n≤100000,1≤m≤200000) ，表示点数和边数。

接下来m 行，每行包含两个正整数ui,vi(1≤ui,vi≤n,ui≠vi) ，表示一条ui→vi 的单向边。

Output

对于每组数据输出一行一个整数，即满足条件的点对数。

Sample Input

1 6 7 1 2 2 3 3 1 4 5 5 6 6 4 2 6

Sample Outpu

6

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
int T,n,m,cnt[2],t,head[2][maxn],vis[maxn],p[maxn],f[maxn];
ll res[maxn];
struct E{
	int to,nex;
}e[2][2*maxn];
int add(int u,int v,int flag){
	e[flag][++cnt[flag]].to=v;
	e[flag][cnt[flag]].nex=head[flag][u];
	head[flag][u]=cnt[flag];
}
int dfs1(int x){
	vis[x]=1;
	for(int k=head[1][x];k;k=e[1][k].nex){
		int v=e[1][k].to;
		if(!vis[v]) dfs1(v);
	}				
	p[++t]=x;
} 
int dfs2(int x,int y){
	vis[x]=0;f[x]=y;
	for(int k=head[0][x];k;k=e[0][k].nex){
		int v=e[0][k].to;
		if(vis[v]) dfs2(v,y);
	}		
}
//dfs1和2  是一正一反图即可，不一定dfs2反1正 
int main(){
	int u,v,i;
	ll ans;
	cin>>T;
	while(T--){
		scanf("%d%d",&n,&m);
		memset(head,0,sizeof(head));//head初始为0 是为了与后面k一致。若初始化为-1则k!=-1 
		memset(res,0,sizeof(res));
		//vis不用重置 dfs2结束后又都为0了 
		cnt[0]=cnt[1]=0; 
		//add正反图时 cnt都得从0开始，所以也要用0，1分别存。不能公用一个
		for(int i=1;i<=m;i++){
			scanf("%d%d",&u,&v);
			add(u,v,0); 
			add(v,u,1);//反图 
		}
		for(t=0,i=1;i<=n;i++){
			if(!vis[i]) dfs1(i);
		}
		for(i=n;i>0;i--){
			if(vis[p[i]]) dfs2(p[i],p[i]);
		}
		for(int i=1;i<=n;i++){
			res[f[i]]++;
			//res不为0 说明该点是一个连通分量的根，res的值为连通分量中点的个数 
		}
		ans=0;
		for(int i=1;i<=n;i++){
			//若为0则不是强连通分量，若为1则没有点对 
			//任意一个点和其余n-1个点均相连，对数还要/2。先除后乘防止爆ll 
			if(res[i]>=2) {
				//判断奇偶，好用偶数除2，防止除漏了 
				if(res[i]%2)
					ans+=(res[i]-1)/2*res[i];
				else
					ans+=res[i]/2*(res[i]-1);			
			}
		} 
		printf("%lld\n",ans);
	}
	return 0;
}