1. 程式人生 > >#103-[最小生成樹之prim演算法]Agri-Net

#103-[最小生成樹之prim演算法]Agri-Net

Description

農夫John被選為他所在市鎮的市長。而他的競選承諾之一是將在該地區的所有的農場用網際網路連線起來。現在他需要您的幫助。

John的農場已經連上了高速連線的網際網路,現在他要將這一連線與其他農場分享。為了減少成本,他希望用最短長度的光纖將他的農場與其他農場連線起來。

給出連線任意兩個農場所需要的光纖長度的列表,請您找到將所有農場連線在一起所需要光纖的最短長度。

任何兩個農場之間的距離不會超過100,000。

Input

      輸入包括若干組測試用例。每組測試用例的第一行給出農場數目N (3<=N<=100)。然後是N´N的鄰接矩陣,其每個元素表示一個農場與另一個農場之間的距離。邏輯上,有N

行,每行有N個空格間隔的整數。一行接一行地輸入,每行不超過80個字元。當然,矩陣的對角線為0,因為一個農場到自己的距離為0。

Output

     對每個測試用例,輸出一個整數,表示把所有農場連線在一起所需要光纖的最短長度。

Sample Input

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

Sample Output

28

稠密圖(幾乎就是完全圖了),prim演算法(炒 雞)適用。

#include <iostream>
#include <cstring>

#define SIZE 110
#define INF 0x3F3F3F3F

using namespace std;

int graph[SIZE][SIZE], lowcost[SIZE];
bool visited[SIZE];

int main(int argc, char** argv)
{
	int n, i, j, res, mincost, index;
	
	while (~scanf("%d", &n))
	{
		memset(visited, false, sizeof (visited));
		res = 0;
		for (i = 1; i <= n; ++i)
		{
			for (j = 1; j <= n; ++j)
			{
				scanf("%d", &graph[i][j]); // prim演算法用的是鄰接矩陣,這裡直接輸入.
			}
		}
		visited[1] = true;
		for (i = 2; i <= n; ++i) // prim演算法求最小生成樹
		{
			lowcost[i] = graph[1][i];
		}
		for (i = 1; i < n; ++i)
		{
			mincost = INF;
			index = -1;
			for (j = 2; j <= n; ++j)
			{
				if ((!visited[j]) && (mincost > lowcost[j]))
				{
					mincost = lowcost[j];
					index = j;
				}
			}
			visited[index] = true;
			res += lowcost[index];
			for (j = 2; j <= n; ++j)
			{
				lowcost[j] = min(lowcost[j], graph[index][j]);
			}
		}
		printf("%d\n", res);
	}
	
	return 0;
}