来自学长的快乐AK题——Day2 B

B——最小生成树

Description

n个人，每个人都有一份情报

人之间都可以交流，交流有对应的费用，交流完后各自享有对方的情报

A与B交流，A与C交流，B与C也可以互相享有

派遣n个人中任意一人去执行任务都有对应的费用

要求：派遣人去执行任务，派遣的人需要包含所有的情报(不一定只派遣一个)，问最小花费

思路

比赛时一眼看过去，就知道遇见了你，我的最小生成树！

首先，考虑两个人花一定费用见面，就能互相知道对方的情报，就联想到用一条边将两个点连接起来。考虑派遣出去的人必须得到所有人的情报，就联想到一张图中所有的点互相连通。
用最小的花费使得所有的点互相连通，这还不是最小生成树吗？

最后，执行任务的费用可以视为图中再加入一个n+1号节点，与每个人新建一条边，边权为这个人执行任务的费用。

对于这个n+1个节点的图跑一遍最小生成树算法即可。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1e6;
struct edge
{
	int u,v,w;
}a[N];
int n,k,cnt,fa[N];
long long ans;
void add(int u,int v,int w)
{
	cnt++;
	a[cnt].u=u;
	a[cnt].v=v;
	a[cnt].w=w;
}
int find(int x)
{
	if(fa[x]==x) return fa[x];
	return fa[x]=find(fa[x]);
}
void join(int x,int y)
{
	int fx=find(x),fy=find(y);
	fa[fx]=fy;
}
bool cmp(edge a,edge b){return a.w<b.w;}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	for(int j=1;j<=n;j++)
	{
		scanf("%d",&k);
		if(i<j) add(i,j,k);
	}
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&k);
		add(n+1,i,k);
	}
	sort(a+1,a+cnt+1,cmp);
	for(int i=1;i<=n+1;i++) fa[i]=i;
	for(int i=1,j=1;i<=n;i++)
	{
		while(find(a[j].u)==find(a[j].v)) j++;
		join(a[j].u,a[j].v);
		ans+=a[j].w;
	}
	printf("%lld",ans);
	return 0;
}