题目大意:
在一个星球上,有n个点,m条边,其中有k个点是不同国家的政府。 要求结点没有到自己的边,任意两个结点间只存在一条边,任意两个政府点之间不存在路径。 问满足上面三个条件下,最多往图中添加几条边。
做法:
codeforces还真够难得,这只是道A题啊!
我们可以先判断满足上述三个条件时,最多能有多少条边。 首先找到每个政府点所在的树的点的个数cnt,产生的边的条数是cnt*(cnt-1)/2。 这样我们就有了一个贪心的做法:把所有的不与任何政府点连通的点全部加到含有政府点的结点最多的树上面去,得到节点数num,产生的边数是num*(num-1)/2 。最后统计结束后,再减点已经存在的边数m就可以了。 统计树的结点数,用并查集处理一下就好了,注意为了方便操作,直接把政府点做为所在树的根节点。
代码如下(调了老半天):
#include<iostream>
#include<cstring>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdio>
#define M 100005
using namespace std;
long long a[M],b[M];
bool vis[M];
int x,y;
int find(int x)
{
if (x==a[x]) return x;
else return a[x]=find(a[x]);
}
void make(int x,int y)
{
int fx=find(x);
int fy=find(y);
if (fx==fy) return;
if (vis[fx]){
b[fx]=b[fx]+b[fy];
a[fy]=fx;
return;
}
else b[fy]=b[fy]+b[fx],a[fx]=fy;
return;
}
int main()
{
int n,m,k;
scanf("%d%d%d",&n,&m,&k);
memset(vis,false,sizeof(vis));
for (int i=1;i<=n;i++) a[i]=i;
for (int i=1;i<=k;i++) scanf("%d",&x),vis[x]=true;
for (int i=1;i<=n;i++) b[i]=1;
for (int i=1;i<=m;i++){
scanf("%d%d",&x,&y);
make(x,y);
}
long long Max1=0; int to;
for (int i=1;i<=n;i++) if (b[i]>Max1&&vis[i]) Max1=b[i],to=i;
for (int i=1;i<=n;i++){
if ((!vis[i]&&find(i)==i)) make(to,i);//如果它是自己独立成为一块的并且没有政府管辖,那么就把它加进去。
}
long long ans=0;
for (int i=1;i<=n;i++) if (vis[i]||i==find(i)) ans=ans+((b[i]-1)*b[i])/2;
printf("%lld",ans-m);
return 0;
}