题目链接
https://www.luogu.org/problemnew/show/P1546
题目背景
农民约翰被选为他们镇的镇长!他其中一个竞选承诺就是在镇上建立起互联网,并连接到所有的农场。当然,他需要你的帮助。
题目描述
约翰已经给他的农场安排了一条高速的网络线路,他想把这条线路共享给其他农场。为了用最小的消费,他想铺设最短的光纤去连接所有的农场。
你将得到一份各农场之间连接费用的列表,你必须找出能连接所有农场并所用光纤最短的方案。每两个农场间的距离不会超过100000
输入输出格式
输入格式:
第一行: 农场的个数,N(3<=N<=100)。
第二行..结尾: 后来的行包含了一个N*N的矩阵,表示每个农场之间的距离。理论上,他们是N行,每行由N个用空格分隔的数组成,实际上,他们限制在80个字符,因此,某些行会紧接着另一些行。当然,对角线将会是0,因为不会有线路从第i个农场到它本身。
输出格式:
只有一个输出,其中包含连接到每个农场的光纤的最小长度。
思路
一道最小生成树的模板题,使用kruskal算法,我们可以定义结构体存边,并进行sort排序,然后按照常规思路做就好了
代码
1 #include<bits/stdc++.h>
2 #include<algorithm>
3 using namespace std; 4 struct node { 5 int x,y,w; 6 } point[10101]; 7 int fat[10101]; 8 int n,x,k; 9 int num=0,ans=0; 10
11 int find(int x) { 12 if(fat[x]==x)return x; 13 return fat[x]=find(fat[x]); 14 } 15
16 void hebing(int x,int y) { 17 int r1=find(x); 18 int r2=find(y); 19 if(r1!=r2)fat[r1]=r2; 20 } 21
22 bool comp(node a,node b) { 23 return a.w<b.w; 24 } 25
26 int main() { 27 scanf("%d",&n); 28 for(int i=1; i<=n; i++) { 29 for(int j=1; j<=n; j++) { 30 scanf("%d",&x); 31 if(x!=0) { 32 num++; 33 point[num].x=i; 34 point[num].y=j; 35 point[num].w=x; 36 } 37 } 38 } 39 for(int i=1; i<=n; i++)fat[i]=i; 40 sort(point+1,point+1+num,comp); 41 for(int i=1; i<=num; i++) { 42 if(find(point[i].x)!=find(point[i].y)) { 43 hebing(point[i].x,point[i].y); 44 ans+=point[i].w; 45 k++; 46 } 47 if(k==n-1)break; 48 } 49 cout<<ans<<'\n'; 50 return 0; 51 }