校园网络[tarjan]

传送门

发现第一问就是入度为0的个数 , 第二问是出度为0 的个数


#include<bits/stdc++.h>
#define N 500050
using namespace std;
int first[N],next[N],to[N],tot,n,in[N],out[N],ans1,ans2;
int dfn[N],low[N],id[N],sta[N],insta[N],cnt,sign,top;
int read(){
	int cnt=0;char ch=0;
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))cnt=cnt*10+(ch-'0'),ch=getchar();
	return cnt;
}
void add(int x,int y){
	next[++tot]=first[x],first[x]=tot,to[tot]=y;
}
void dfs(int u){
	dfn[u]=low[u]=++sign;
	sta[++top]=u,insta[u]=1;
	for(int i=first[u];i;i=next[i]){
		int t=to[i]; 
		if(!dfn[t]) dfs(t),low[u]=min(low[u],low[t]);
		else if(insta[t] && dfn[t]<low[u]) low[u]=dfn[t];
	}
	if(low[u]==dfn[u]){
		cnt++; do{
			id[sta[top]]=cnt,insta[sta[top]]=0;
		}while(sta[top--]!=u);
	}
}
int main(){
	n=read();
	for(int i=1;i<=n;i++){
		int x=read(); while(x!=0){
			add(i,x); x=read(); 
		}
	}
	for(int i=1;i<=n;i++) if(!dfn[i]) dfs(i);
	for(int i=1;i<=n;i++)
		for(int j=first[i];j;j=next[j])
			if(id[to[j]]!=id[i]) in[id[to[j]]]++,out[id[i]]++;
	for(int i=1;i<=cnt;i++){
		if(!in[i]) ans1++; if(!out[i]) ans2++;
	}
	printf("%d\n%d",ans1,ans2); return 0;
}

猜你喜欢

转载自blog.csdn.net/sslz_fsy/article/details/84671440