CF744A Hongcow Builds A Nation 题解

题目：CF744A Hongcow Builds A Nation

图论-连通块-数论/数学-排列组合

题目大意

给出一张 $n$ 个点， $m$ 条边的无向图，其中有$k$个顶点是警察局
求最多能多加多少条边，使得加完边后的图 无重边，无自环，且一个连通块里最多只有一个警察局

对于每一个连通块，我们先要让它的边数最大化：设节点数为 $x$，已有的边数为 $y$ ，则还能多加的边数就是 $\frac{x(x-1)}{2}-y$，这样就把每一个连通块边数加到极致

利用贪心的思想，将所有连通块按大小降序排列，每次合并最大的两个连通块，前提是合并后连通块内不会超过一个警察局

$Tips:$ 合并两个大小分别为 $x,y$ 的连通块，可以多加的边数为 $x\times y$

#include<cstdio>
#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
const int Maxn=1010,inf=0x3f3f3f3f;
vector <int> e[Maxn],bin;
int f[Maxn],c[Maxn],sum[Maxn];
bool vis[Maxn];
int n,m,k,ans;
inline bool cmp(int x,int y)
{
	return c[x]>c[y];
}
inline int read()
{
	int s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0' && ch<='9')s=(s<<3)+(s<<1)+(ch^48),ch=getchar();
	return s*w;
}
void bfs(int s,int fa)
{
	queue <int> q;
	f[s]=fa,q.push(s);
	while(q.size())
	{
		int x=q.front();
		q.pop();
		vis[fa] |= vis[x];
		++c[fa];
		for(int i=0;i<e[x].size();++i)
		{
			int y=e[x][i];
			++sum[fa];
			if(f[y])continue;
			f[y]=fa,q.push(y);
		}
	}
	sum[fa]>>=1;
}
int main()
{
	//freopen("in.txt","r",stdin);
	n=read(),m=read(),k=read();
	for(int i=1;i<=k;++i)
	{
		int x=read();
		vis[x]=1;
	}
	
	for(int i=1;i<=m;++i)
	{
		int x=read(),y=read();
		e[x].push_back(y);
		e[y].push_back(x);
	}
	
	for(int i=1;i<=n;++i)
	{
		if(f[i]>0)continue;
		bfs(i,i);
//		printf("heehehe %d %d %d\n",f[i],c[i],sum[i]);
		ans+=(c[i]*(c[i]-1)>>1)-sum[i];
		if(f[i]==i)bin.push_back(i);
	}
	
	sort(bin.begin(),bin.end(),cmp);
	for(int i=0;i<bin.size();++i)
	{
		for(int j=i+1;j<bin.size();++j)
		{
			int x=bin[i],y=bin[j];
			
			if(vis[f[x]] && vis[f[y]])continue;
//			printf("%d %d\n",f[x],f[y]);
//			printf("%d %d\n",c[f[x]],c[f[y]]);
//			printf("%d\n",c[f[x]]*c[f[y]]);
//			puts("");
			ans+=c[f[x]]*c[f[y]];
			c[f[x]]+=c[f[y]];
			vis[f[x]] |= vis[f[y]];
			f[f[y]]=f[x];
		}
	}
	
	printf("%d\n",ans);
	return 0;
}