soj 2012. King

http://soj.me/show_problem.php?pid=2012

KJ推荐的这道题，关于强联通分量的。这里主要是利用 Kosaraju 算法，

两次bfs求出强联通分量，第二次dfs2()主要是以第一次的ord的倒序进行dfs。

缩点，然后把原图变为一个新的DAG图，然后统计新的DAG图的入点为0的个数。

注意这里输出的是序号，不是个数。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <string.h>
#include <cmath>
using namespace std;

const int maxn = 1015;
vector<int>adj[maxn];
vector<int>radj[maxn];
vector<int>DAG_adj[maxn];
vector<int>ord;
int vis[maxn],g[maxn];
int n,cnt;
char map[maxn][maxn];
int num[maxn];
int innum[maxn];
int outnum[maxn];
void dfs1(int u)
{
   vis[u]=true;
   for(int i=0;i<adj[u].size();i++)
   {
	   int v=adj[u][i];
	   if(!vis[v])
		   dfs1(v);
   }
   ord.push_back(u);
}

void dfs2(int u)
{
	vis[u]=true;
	g[u]=cnt;
	for(int i=0;i<radj[u].size();i++)
	{
		int v=radj[u][i];
		if(!vis[v])
			dfs2(v);
	}
}
void Kosaraju()
{
	 ord.clear();
	 memset(vis,false,sizeof(vis));
	 for(int i=1;i<=n;i++)
	 {
		 if(!vis[i])
			 dfs1(i);
	 }
	 memset(vis,false,sizeof(vis));
	 cnt=0;
	 for(int i=ord.size()-1;i>=0;i--)
	 {
		 if(!vis[ord[i]])
		 {
			 cnt+=1;
			 dfs2(ord[i]);
		 }
	 }
}

int main()
{
    while(scanf("%d",&n)!=EOF)
	{
		for(int i=1;i<=n;i++)
		{
			adj[i].clear();
			radj[i].clear();
			DAG_adj[i].clear();
			g[i]=-1;
		}
		for(int i=1;i<=n;i++)
			 scanf("%s",map[i]+1);

		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				if(map[i][j]=='1')
				{
                   adj[i].push_back(j);
				   radj[j].push_back(i);
				}
			}
		}

		Kosaraju();
		memset(num,0,sizeof(num));
		for(int i=1;i<=n;i++)
			num[g[i]]+=1;
        for(int i=1;i<=n;i++)
		{
			for(int j=0;j<adj[i].size();j++)
			{
				int u=adj[i][j];
				if(g[i]!=g[u])
					DAG_adj[g[i]].push_back(g[u]);
			}
		}

		memset(innum,0,sizeof(innum));
		memset(outnum,0,sizeof(outnum));

		for(int i=1;i<=cnt;i++)
		{
			for(int j=0;j<DAG_adj[i].size();j++ )
			{
				int u=DAG_adj[i][j];
				innum[u]+=1;
				outnum[i]+=1;
			}
		}

		int zeronum=0;
		int ok=0;
		for(int i=1;i<=cnt;i++)
		{
			if(innum[i]==0)
			{
			   zeronum+=1;
               ok=i;
			}
			if(zeronum>=2)
				break;
		}
		if(zeronum>=2)
			printf("-1\n");
		else
		{
			for(int i=1;i<=n;i++)
			{
				if(g[i]==ok)
				{
					printf("%d\n",i);
					break;
				}
			}
		}
	}
}