算法7-9：最小生成树

   

最小生成树问题是实际生产生活中十分重要的一类问题。假设需要在n个城市之间

建立通信联络网，则连通n个城市只需要n-1条线路。这时，自然需要考虑这样

一个问题，即如何在最节省经费的前提下建立这个通信网。

可以用连通网来表示n个城市以及n个城市之间可能设置的通信线路，其中网的顶

点表示城市，边表示两个城市之间的线路，赋于边的权值表示相应的代价。对于n

个顶点的连通网可以建立许多不同的生成树，每一棵生成树都可以是一个通信网

。现在，需要选择一棵生成树，使总的耗费最小。这个问题就是构造连通网的最小代

价生成树，简称最小生成树。一棵生成树的代价就是树上各边的代价之和。

而在常用的最小生成树构造算法中，普里姆（Prim）算法是一种非常常用的算法。以

下是其算法的大致结构：

在本题中，读入一个无向图的邻接矩阵（即数组表示），建立无向图并按照以上描述

中的算法建立最小生成树，并输出最小生成树的代价。

输入的第一行包含一个正整数n，表示图中共有n个顶点。其中n不超过50。

以后的n行中每行有n个用空格隔开的整数，对于第i行的第j个整数，如果不为0，则表示第i个顶点和第j个顶点有直接连接且代价为相应的值，0表示没有直接连接。当i和j相等的时候，保证对应的整数为0。

输入保证邻接矩阵为对称矩阵，即输入的图一定是无向图，且保证图中只有一个连通分量。

只有一个整数，即最小生成树的总代价。请注意行尾输出换行。

4

0 2 4 0

2 0 3 5

4 3 0 1

0 5 1 0

6

     

---

#include<iostream>//#include<bits/stdc++.h>
using namespace std;//标准命名空间 

const int INF = 100000;
const int MAXN = 55;

int a[MAXN][MAXN]={0},n;
//int vis[MAXN]={0};
int d[MAXN][MAXN];
int sum=0;

int included[MAXN];

bool inclu(int t)
{
	for (int i = 1; i < MAXN; i++)
	{
		if (t == included[i])return true;
	}return false;
}

void MST()
{
	int k = 1;
	int x = 1;//从任意一个点出发，这里从第一个出发为例
	included[k] = x; k++;//把x结点归入集合
	int t = n - 1;
	int m=1;
	while (t--)
	{
		int currm = INF;
		for (int i = 1; i <= n; i++)//寻找下一个要归入的结点
		{
			for (int j = 1; j <= n; j++)
			{
				if (inclu(i) && !inclu(j) && d[i][j] < currm)//i已被归入且j没有被归入
				{					
					currm = d[i][j];
					m = j;					
				}
			}
		}
		
		sum += currm;		
		included[k++] = m;
		//cout << "此次归入长度为" << currm << "的路径" << endl;
		
	}
	cout << sum << endl;
	//cout << k << endl;

}

int main()
{
	cin >> n;
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= n; j++)
		{
			cin >> a[i][j];
			if (a[i][j] == 0)d[i][j] = INF;
			else d[i][j] = a[i][j];
		}
	}
	/*for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= n; j++)
		{
			cout << d[i][j]<<"       ";
		}cout << endl;
	}*/
	
	MST();
		
	return 0;
}