最小生成树（Prim）

思路（贪心）：从单一顶点开始，普里姆算法按照以下步骤逐步扩大树中所含顶点的数目，直到遍及连通图的所有顶点。

1. 输入：一个加权连通图，其中顶点集合为V，边集合为E；
2. 初始化：V_new = {x}，其中x为集合V中的任一节点（起始点），E_new = {}；
3. 重复下列操作，直到V_new = V：
   1. 在集合E中选取权值最小的边（u, v），其中u为集合V_new中的元素，而v则不是（如果存在有多条满足前述条件即具有相同权值的边，则可任意选取其中之一）；
   2. 将v加入集合V_new中，将（u, v）加入集合E_new中；
4. 输出：使用集合V_new和E_new来描述所得到的最小生成树。

其实就是每加入一个顶点，就更新它与未加入的每个顶点的边权值（有更小的边权才更新）；

 

#include <cstdio>
#include <cstring>
#define SIZE 1000
#define MAXCOST 100000000
int graph[100][100];

/**初始化图的所有边为MAXCOST，即正无穷大*/
void initi(int n){
	for(int i = 0;i <= n;i++)
		for(int j = 0;j <= n;j++)
			graph[i][j] = MAXCOST;
}

/**minCost[i]数组表示以第i个顶点为终点的边的最小权值*/ 
int Prim(int m){
	int i, j, vis[SIZE];
	int minCost[SIZE], minW = 0;
	memset(vis, 0, sizeof(vis));
	/**vis[i]=1表示i顶点已加入最小生成树中*/
	vis[1] = 1;
	minCost[1] = 0;
	/**初始化，更新与起点的权值*/
	for(i = 2;i <= m;i++)
		minCost[i] = graph[1][i]; 
	for(i=2;i <= m;i++){
		int min = MAXCOST, minLoc=0;
		/**找出当前最小树中的顶点与未加入顶点中权值最小的边*/ 
		for(j=2;j<=m;j++)
			if(!vis[j]&&minCost[j] < min){
				minLoc = j;
				min = minCost[j];
			}
		/**将当前找到构成最小边的顶点加入树中*/
		vis[minLoc] = 1;
		minW += min;
		/**更新最小树中与未加入顶点的权值*/ 
		for(j = 2;j <= m;j++)
			if(!vis[j]&&minCost[j] > graph[minLoc][j])
				minCost[j] = graph[minLoc][j];
	}
	return minW;		
}

int main(){
	int n, m, i, j, u, v, w;
	/**n为边数，m为顶点个数*/
	scanf("%d%d", &n, &m);
	initi(m);
	for(i = 0;i < n;i++){
		scanf("%d%d%d", &u, &v, &w);
		graph[u][v] = graph[v][u] = w;
	}
	printf("%d\n", Prim(m));
	return 0;
}