题目传送门:https://www.luogu.org/problemnew/show/P1546
题目背景
农民约翰被选为他们镇的镇长!他其中一个竞选承诺就是在镇上建立起互联网,并连接到所有的农场。当然,他需要你的帮助。
题目描述
约翰已经给他的农场安排了一条高速的网络线路,他想把这条线路共享给其他农场。为了用最小的消费,他想铺设最短的光纤去连接所有的农场。
你将得到一份各农场之间连接费用的列表,你必须找出能连接所有农场并所用光纤最短的方案。每两个农场间的距离不会超过100000
输入输出格式
输入格式:
第一行: 农场的个数,N(3<=N<=100)。
第二行..结尾: 后来的行包含了一个N*N的矩阵,表示每个农场之间的距离。理论上,他们是N行,每行由N个用空格分隔的数组成,实际上,他们限制在80个字符,因此,某些行会紧接着另一些行。当然,对角线将会是0,因为不会有线路从第i个农场到它本身。
输出格式:
只有一个输出,其中包含连接到每个农场的光纤的最小长度。
题解:
在一个图中找出一个子图,使它包含所有节点,并且边权之和最小。
很明显,这道题是一个最小生成树的模板题。
(最小生成树有两种算法,Prim和Kruskal,本题解用的是Kruskal)
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int N = 105;
const int M = 10050;
struct Edge
{
int u,v,w;
}edge[M];
int cc;
int n;
int ans=0;
int fa[N];
bool cmp(Edge x,Edge y)
{
return (x.w<y.w);
}
int getf(int x)
{
if(fa[x]!=x)
fa[x]=getf(fa[x]);
return fa[x];
}
int main()
{
int i,j,k;
cin>>n;
for(i=1;i<=n;i++)
{
for(j=1;j<=n;j++)
{
cc++;
edge[cc].u=i;
edge[cc].v=j;
cin>>edge[cc].w;
}
}
sort(edge+1,edge+cc+1,cmp);
for(i=1;i<=n;i++)
fa[i]=i;
for(i=1;i<=cc;i++)
{
int a=getf(edge[i].u);
int b=getf(edge[i].v);
if(a!=b)
{
fa[a]=b;
ans+=edge[i].w;
}
}
cout<<ans<<endl;
return 0;
}