Tarjan縮點——受歡迎的牛

題目來源

洛谷P2341[HAOI2006]受歡迎的牛

https://www.luogu.org/problem/show?pid=2341

思路

以奶牛爲點、仰慕關係爲邊建立原圖

用Tarjan算法求出原圖中所有點所屬的SCC

以剛求出的SCC爲點、原圖中連接位於不同SCC中的點的邊爲邊建新圖

如果新圖中有且只有一個點無出度 則該點對應原圖的SCC中的所有奶牛都是明星奶牛 該SCC所包含的點數即爲所求答案

否則 原圖中無明星奶牛

代碼（C++)

#include <stack>
#include <cstdio>
#include <bitset>
using namespace std;
bitset <200010> g,in;	stack<int> s;
int n,m,u,v,cnt=0,num,t=0;
struct way{int en,ne;}p[100010];
int he[20010],dfn[10010],low[10010];
int scc[20010],sum[20010],ans=0;
void dfs(int pos);
inline void add();
void search(int pos);
int main()
{
	scanf("%d%d",&n,&m); 
	for(int i=1;i<=m;++i)
		scanf("%d%d",&u,&v),add();
	num=n;	g=0;
	for(int i=1;i<=n;++i)
		if(g[i]==0)
			search(i);
	g=0;
	for(int i=1;i<=n;++i)
		if(g[i]==0)
			dfs(i);
	for(int i=1;i<=n;++i)
		++sum[scc[i]];
	for(int i=n+1;i<=num;++i)
	{
		if(he[i]==0)
		{
			if(ans==0)
				ans+=sum[i];
			else
			{
				printf("0");
				return 0;
			}
		}
	}	
	printf("%d",ans);
	return 0;
}
void dfs(int pos)
{
	g[pos]=1;
	int k=he[pos];
	while(k!=0)
	{
		if(scc[pos]!=scc[p[k].en])
		{
			if(g[p[k].en]==0)
				dfs(p[k].en);
			u=scc[pos];
			v=scc[p[k].en];
			add();
		}
		k=p[k].ne;
	}
	return ;
}
void search(int pos)
{
	g[pos]=1;	s.push(pos);
	dfn[pos]=++t;	low[pos]=dfn[pos];
	int k=he[pos];
	while(k!=0)
	{
		if(g[p[k].en]==0)
		{
			search(p[k].en);
			if(low[p[k].en]<low[pos])
				low[pos]=low[p[k].en];
		}
		else if(dfn[p[k].en]<low[pos]&&scc[p[k].en]==0)
			low[pos]=dfn[p[k].en];
		k=p[k].ne;
	}
	if(low[pos]==dfn[pos])
	{
		++num;
		while(s.top()!=pos)
			scc[s.top()]=num,s.pop();
		scc[pos]=num;
		s.pop();
	}	
}
inline void add()
{
	++cnt;
	p[cnt].en=v;
	p[cnt].ne=he[u];	
	he[u]=cnt;
}