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;

				}

			}

		}

	}

}