洛谷 P1546 最短网络 Agri-Net
题目链接
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<<'
';
50 return 0;
51 }