L2-031 深入虎穴（并查集）

题目链接

 

#include<bits/stdc++.h>
using namespace std;
int f[100010],m[100010],v[100010];
int find(int x)
{	
	if(f[x]!=x) 
	{
		int k=f[x];
		f[x]=find(f[x]);
		m[x]+=m[k];
	}
	return f[x];
}
void merge(int x,int y)
{
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy)
	{
		f[fx] = fy;
		m[fx] = m[y] - m[x] + 1;
	} 
}
int main()
{
	int n,t,a;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		f[i]=i;
	}
	for(int i=1;i<=n;i++)
	{
		cin>>t;
		while(t--) 
		{
			cin>>a;
			v[a]=1;
			merge(a,i);
		}
	}
	int r;//入口
	for(int i=1;i<=n;i++)
	if(v[i]==0) {
		r=i;
		break;
	}
	
	int id=0,ma=-1;
	for(int i=1;i<=n;i++)
	{
		if(find(i)==r) 
		{
			if(m[i]>ma)
			ma=m[i],id=i;	
		}
	}
	cout<<id;
	
	return 0;
}