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

题目背景

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

题目描述

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

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

输入格式

第一行农场的个数 N（3≤N≤100）。

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

输出格式

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

输入输出样例

输入 #1

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

输出 #1

28

分析

Prim算法求解最小生成树

• prim算法

Prim算法是一种用于构建最小生成树的贪心算法。它的基本思想是从一个点开始，每次选择一个与当前生成树距离最小的点加入生成树，直到生成树包含了所有的点

Prim算法的基本步骤：

1. 选择一个起始点，将其加入生成树中。
2. 对于生成树中的每个点，找到与其相邻的未加入生成树的点中距离最小的那个点，将其加入生成树中。
3. 重复步骤2，直到生成树包含了所有的点

• 代码具体分析

定义常量INF，表示无穷大，用于初始化d数组。

接下来定义了一个二维数组a，表示图的邻接矩阵，以及一个一维数组d，表示每个节点到最小生成树的距离

在主函数中，首先输入图的节点数n和邻接矩阵a，然后调用prim函数求解最小生成树，并输出最小生成树的权值和mst

prim函数中，首先初始化d数组，然后进行n次循环，每次找到距离最小的未访问节点k，将其标记为已访问，更新最小生成树的权值和mst，然后更新与k相邻的未访问节点的距离d。最后返回mst即可

代码

#include<bits/stdc++.h>//万能头文件，包含了STL中所有的头文件，可以方便地使用STL中的各种数据结构和算法
#define INF 0x3f3f3f3f
using namespace std;
const int N=1001;
bool vis[N];
int a[N][N],d[100001]/*表示每个节点到最小生成树的距离*/,mst,n;
void prim(){
	
	memset(d,INF,sizeof(d));
	d[1]=0;
	for(int i=1;i<=n;i++){
		int k=0;
		for(int j=1;j<=n;j++){
			if(!vis[j]&&d[j]<d[k]) k=j;
		}
		vis[k]=true;
		mst+=d[k];
		for(int j=1;j<=n;j++){
			if(!vis[j]&&a[k][j]<d[j]) d[j]=a[k][j];
		}
	}
	
}
int main(){
	
	cin>>n;
	for(int i=1;i<=n;i++){//输入 
		for(int j=1;j<=n;j++){
			cin>>a[i][j];
		}
	}
	prim();
	cout<<mst<<endl;
	
	
	
	return 0;
}