CF 738div2 D2. Mocha and Diana (Hard Version) 并查集

D2. Mocha and Diana (Hard Version)

 题意：给出大小为n边数分别为m1和m2的森林，求在两个森林中添加边的最大数目（添加边需要满足添加该边后还是森林，并且两个森林中该边不存在）

先说策略，我们选择节点1作为主联通块，然后开始连边，另外一个点选两森林中都不属于主联通块的节点，没有可以连了，我们就选择不在主联通块上的点，将这类点互相连起来。（只能连不同森林不在主联通块上的点）。

根据题意我们可以得知最后的情况，两森林中至少一个森林，所有点都联通。如果两个森林都存在联通块的话，那么一定可以在两森林中找到能连接的点，因为一个点最多可以和n-1个点相连，当图中存在联通块的情况下，它至少和一个点不相连，如果两森林中这点相同，便可以直接连接，如果不同，便可以把不同的点连起来，直到有个森林不能再添加边了。

代码如下

#include
#include

using namespace std;

const int N=1e5+5;

int n,m1,m2,fa1[N],fa2[N];
vector v[3];

int find(int fa[],int x)
{
	if(fa[x]!=x) fa[x]=find(fa,fa[x]);
	return fa[x];
}

int main()
{
	scanf("%d%d%d",&n,&m1,&m2);
	for(int i=1;i<=n;i++)
	{
		fa1[i]=i;
		fa2[i]=i;
	}
	int a,b;
	for(int i=1;i<=m1;i++)
	{
		scanf("%d%d",&a,&b);
		fa1[find(fa1,a)]=fa1[find(fa2,b)];
	}
	for(int i=1;i<=m2;i++)
	{
		scanf("%d%d",&a,&b);
		fa2[find(fa2,a)]=fa2[find(fa2,b)];
	}
	for(int i=2;i<=n;i++)
	{
		if(find(fa1,1)!=find(fa1,i)&&find(fa2,1)!=find(fa2,i))
		{
			v[1].push_back(1);
			v[2].push_back(i);
			fa1[find(fa1,i)]=fa1[find(fa1,1)];
			fa2[find(fa2,i)]=fa2[find(fa2,1)];
		}
	}
	for(int i=2;i<=n;i++)
	{
		if(fa1[i]==i&&fa1[i]!=find(fa1,1)) v[1].push_back(i);
		if(fa2[i]==i&&fa2[i]!=find(fa2,1)) v[2].push_back(i);
	}
	int cnt=min(v[1].size(),v[2].size());
	printf("%d\n",cnt);
	for(int i=0;i