【bzoj4243】交朋友 bfs

       显然两个国家如果能进行会议那么一定要进行会议。考虑怎样的国家可以进行会议，注意到一个点，对于它的出边到达的点（称为后继），如果后继的个数>=2，那么这些后继之间可以两两进行会议。那么不妨将这些点缩成一个点，然后不断缩点直到不能缩为止，这时，对于一个缩起来的点，里面边的数量为包含点的数量*（包含点的数量-1），对于普通的点，对答案的影响即出度。

       不妨用并查集来缩点。注意对于一对点(i,j)存在i->j，j->i，那么也可以把(i,j)缩起来；另外考虑一个缩起来的点x，以及任意一个后继y，显然可以把x和y缩到一起。那么用一个队列存储已经缩起来且还没有更新过后继的点，然后每次取出一个把它的后继和它缩在一起，然后把后继扔进队列（如果已经入过队就无视）。

AC代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#define N 100005
using namespace std;

int n,m,tot,fa[N],sz[N],pnt[N<<2],nxt[N<<2],h[N],od[N]; bool bo[N];
int read(){
	int x=0; char ch=getchar();
	while (ch<'0' || ch>'9') ch=getchar();
	while (ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }
	return x;
}
struct graph{
	int fst[N];
	void add(int x,int y){
		pnt[++tot]=y; nxt[tot]=fst[x]; fst[x]=tot;
	}
}g1,g2;
int getfa(int x){
	return (x==fa[x])?x:fa[x]=getfa(fa[x]);
}
void merge(int x,int y){
	int u=getfa(x),v=getfa(y); if (sz[u]>sz[v]) swap(u,v);
	if (u!=v){ fa[u]=v; sz[v]+=sz[u]; }
}
int main(){
	n=read(); m=read(); int i,x,y;
	for (i=1; i<=m; i++){
		x=read(); y=read(); od[x]++;
		g1.add(x,y); g2.add(y,x);
	}
	for (i=1; i<=n; i++){ sz[i]=1; fa[i]=i; }
	for (i=1; i<=n; i++){
		for (x=g1.fst[i]; x; x=nxt[x]) bo[pnt[x]]=1;
		for (x=g2.fst[i]; x; x=nxt[x])
			if (bo[pnt[x]]) merge(i,pnt[x]);
		for (x=g1.fst[i]; x; x=nxt[x]) bo[pnt[x]]=0;
	}
	for (i=1; i<=n; i++){
		for (x=g1.fst[i],y=0; x; x=nxt[x])
			if (y) merge(pnt[x],y); else y=pnt[x];
	}
	int head=0,tail=0; memset(bo,1,sizeof(bo));
	for (i=1; i<=n; i++)
		if (sz[getfa(i)]>1){ bo[i]=0; h[++tail]=i; }
	while (head<tail){
		x=h[++head];
		for (i=g1.fst[x]; i; i=nxt[i]){
			y=pnt[i]; merge(x,y);
			if (bo[y]){ bo[y]=0; h[++tail]=y; }
		}
	}
	long long ans=0;
	for (i=1; i<=n; i++)
		if (getfa(i)==i)
			if (sz[i]>1) ans+=(long long)sz[i]*(sz[i]-1);
			else ans+=od[i];
	printf("%lld\n",ans);
	return 0;
}

by lych

2016.3.26