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
,sz
,pnt[N<<2],nxt[N<<2],h
,od
; bool bo
;
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
;
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