D2. Mocha and Diana (Hard Version)
题意:给出大小为n边数分别为m1和m2的森林,求在两个森林中添加边的最大数目(添加边需要满足添加该边后还是森林,并且两个森林中该边不存在)
先说策略,我们选择节点1作为主联通块,然后开始连边,另外一个点选两森林中都不属于主联通块的节点,没有可以连了,我们就选择不在主联通块上的点,将这类点互相连起来。(只能连不同森林不在主联通块上的点)。
根据题意我们可以得知最后的情况,两森林中至少一个森林,所有点都联通。如果两个森林都存在联通块的话,那么一定可以在两森林中找到能连接的点,因为一个点最多可以和n-1个点相连,当图中存在联通块的情况下,它至少和一个点不相连,如果两森林中这点相同,便可以直接连接,如果不同,便可以把不同的点连起来,直到有个森林不能再添加边了。
代码如下
#include<iostream>
#include<vector>
using namespace std;
const int N=1e5+5;
int n,m1,m2,fa1[N],fa2[N];
vector<int> 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<cnt;i++)
{
printf("%d %d\n",v[1][i],v[2][i]);
}
}