牛客寒假练习 4.E 通知小弟 强连通缩点,DAG

题意:有一个头头,它有m个联系人,当它打电话给某个人u时,u会将消息传递给u所有的联系人.
总共有n个人 问头头最少要发给多少个联系人 才能使n个人都能接受到消息.
0<=a[i]<n,m<=500


建立一个有向图,有环的只要通知环上的任意一个点即可,先将图中强联通分量缩成点,原图变为DAG.

DAG中度数为0的肯定要通知,度数不为0的点不断往前回溯肯定能到某个度数为0的点. 则答案为度数为0的顶点个数.

#include <bits/stdc++.h>
using namespace std;
const int N=5e2+5;
int n,m,mk[N],in[N],vis[N],can[N];
vector<int> e[N];
int dfn[N],low[N],ins[N],bel[N],scc_cnt,dfs_id;
stack<int> st; 
void init()
{
	for(int i=1;i<=n;i++)
		e[i].clear();
	memset(can,0,sizeof(can));
	memset(vis,0,sizeof(vis));
	memset(in,0,sizeof(in));
	memset(dfn,0,sizeof(dfn));
	memset(ins,0,sizeof(ins));
	scc_cnt=dfs_id=0;
}
void tarjan(int u)
{
	dfn[u]=low[u]=++dfs_id;
	st.push(u),ins[u]=1;
	for(int i=0;i<e[u].size();i++)
	{
		int v=e[u][i];
		if(!dfn[v])
		{
			tarjan(v);
			low[u]=min(low[u],low[v]);
		}
		else if(ins[v])
			low[u]=min(low[u],dfn[v]);
	}
	if(dfn[u]==low[u])
	{
		scc_cnt++;
		while(true)
		{
			int v=st.top();
			st.pop();
			bel[v]=scc_cnt;
			if(v==u)
				break;
		}
	}
}
int main()
{
	while(~scanf("%d%d",&n,&m))
	{
		init(); 
		int u,x;
		for(int i=0;i<m;i++)
			scanf("%d",&u),mk[u]=1;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&x);
			for(int j=0;j<x;j++)
				scanf("%d",&u),e[i].push_back(u);
		}
		for(int i=1;i<=n;i++)
			if(!dfn[i])
				tarjan(i);
		for(int i=1;i<=n;i++)
		{
			for(int j=0;j<e[i].size();j++)
			{
				int v=e[i][j];
				if(bel[i]!=bel[v])
					in[bel[v]]++;
			}
		}
		for(int i=1;i<=n;i++)
			if(mk[i])
				can[bel[i]]=1;
		int ans=0,flag=true;
		for(int i=1;i<=n;i++)
		{
			if(vis[bel[i]]||in[bel[i]]>0)	continue;
			if(can[bel[i]]==false)	flag=false;
			ans++,vis[bel[i]]=1;
		}
		if(!flag)
			ans=-1;
		printf("%d\n",ans);
	}
	return 0;
}