【洛谷】P1546 [USACO3.1]最短网络 Agri-Net C++（Kruskal+并查集）

【洛谷】P1546 [USACO3.1]最短网络 Agri-Net C++（Kruskal+并查集）

[USACO3.1]最短网络 Agri-Net

题目背景

Farmer John 被选为他们镇的镇长！他其中一个竞选承诺就是在镇上建立起互联网，并连接到所有的农场。当然，他需要你的帮助。

题目描述

FJ 已经给他的农场安排了一条高速的网络线路，他想把这条线路共享给其他农场。为了用最小的消费，他想铺设最短的光纤去连接所有的农场。

你将得到一份各农场之间连接费用的列表，你必须找出能连接所有农场并所用光纤最短的方案。每两个农场间的距离不会超过 $10^5$。

输入格式

第一行农场的个数 $N$（$3\le N\le 100$）。

接下来是一个 $N\times N$ 的矩阵，表示每个农场之间的距离。理论上，他们是 $N$ 行，每行由 $N$ 个用空格分隔的数组成，实际上，由于每行 $80$ 个字符的限制，因此，某些行会紧接着另一些行。当然，对角线将会是 $0$，因为不会有线路从第 $i$ 个农场到它本身。

输出格式

只有一个输出，其中包含连接到每个农场的光纤的最小长度。

样例 #1

样例输入 #1

4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0

样例输出 #1

28

提示

题目翻译来自NOCOW。

USACO Training Section 3.1

代码+注释如下：如有不理解，欢迎评论区讨论！

#include<bits/stdc++.h>
using namespace std;
/*
	初始化父节点、将各条边排序、
	循环每条边（判断构成此边的两节点是否属于同一模块，不属于，则合并，将边权加入sum）
	反复循环，直到遍历完所有边/加入的边等于结点数-1为止 
*/
const int M = 100002;

//每条边的各类属性值以结构体展现
struct edge
{
	int x;
	int y;
	int wi;
}E[M]; 

//存放各自父节点
int f[M];

//寻找父节点，并进行路径压缩
int find(int x)	
{
	if(x!=f[x])
	{
		f[x] = find(f[x]); 
	}
	return f[x];
} 

//sort后为升序序列
bool cmp(edge x1,edge x2){
	return x1.wi<x2.wi;
}

int main(){
	int n,m=0;
	cin>>n;
	int k;
	for(int i=1;i<=n;i++)
	{
		f[i] = i;
		for(int j=1;j<=n;j++)
		{
			cin>>k;
			if(j>i)//因为对称矩阵，检索上半部分即可
			{
				m++;
				E[m].x = i;
				E[m].y = j;
				E[m].wi = k;
			}
		}
	}
	//排序（起始地址、末地址、cmp函数）
	sort(E+1,E+m+1,cmp);
	int ans = 0,num = 1;
	for(int t=1;t<=m;t++)
	{
		//如若不属于同一集合，将其合并
		if(find(E[t].x)!=find(E[t].y))
		{
			f[find(E[t].x)] = E[t].y;
			ans = ans + E[t].wi;
			//一条边加入
			num++;
			//边数等于顶点数-1，即结束
			if(num == n)
			{
				break;
			}
		}
	}
	
	cout<<ans;
	
	return 0;
}