POJ 2762 Going from u to v or from v to u?（强连通分量）

题目链接：http://poj.org/problem?id=2762

题意：给定一个有向图。问这个图是否满足下面的性质：对于图中任意两个顶点u和v，存在u到v的路径或者v到u的路径.

思路：步骤如下：

（1）求强连通分量；

（2）缩点：将原图换成新图；

（3）在新图中拓扑排序，若每次只有一个入度为0的点，则满足题意。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

struct Node
{
int v,next;
};

Node edges[6005];
int head[1005],e;
int n,m,C;
int st[1005],top,dfn[1005],low[1005],ID,num,visit[1005];
int p[1005];

void Add(int u,int v)
{
edges[e].v=v;
edges[e].next=head[u];
head[u]=e++;
}

void DFS(int u)
{
ID++;
dfn[u]=low[u]=ID;st[top++]=u;visit[u]=1;
int i,v;
for(i=head[u];i!=-1;i=edges[i].next)
{
v=edges[i].v;
if(!dfn[v])
{
DFS(v);
low[u]=min(low[u],low[v]);
}
else if(visit[v]) low[u]=min(low[u],dfn[v]);
}
if(low[u]>=dfn[u])
{
num++;
do
{
v=st[top-1];
visit[v]=0;
p[v]=num;
top--;
}while(v!=u);

}
}

int g[1005][1005];
int deg[1005];

int OK()
{
int i,j,u,v,x;
memset(g,0,sizeof(g));
memset(deg,0,sizeof(deg));
for(u=1;u<=n;u++)
{
for(i=head[u];i!=-1;i=edges[i].next)
{
v=edges[i].v;
if(p[u]!=p[v]&&!g[p[u]][p[v]])
{
g[p[u]][p[v]]=1;
deg[p[v]]++;
}
}
}
for(i=1;i<num;i++)
{
x=0;
for(j=1;j<=num;j++) if(deg[j]==0) x++,u=j;
if(x>1) return 0;
for(v=1;v<=num;v++) if(g[u][v]) deg[v]--;
deg[u]=-1;
}
return 1;
}

int main()
{
for(scanf("%d",&C);C--;)
{
scanf("%d%d",&n,&m);
memset(head,-1,sizeof(head));
e=0;
int u,v,i;
for(i=1;i<=m;i++)
{
scanf("%d%d",&u,&v);
Add(u,v);
}
top=0;ID=0;num=0;
memset(dfn,0,sizeof(dfn));
memset(visit,0,sizeof(visit));
for(i=1;i<=n;i++) if(!dfn[i]) DFS(i);
if(OK()) puts("Yes");
else puts("No");
}
return 0;
}