最小生成树

Description

在一张图上有N个点，点与点之间的连接的花费都已经告诉你了，请你设计一下，如何解决这个“最小生成树”的问题。要求用prim方法求解。

Input

首先输入一个数字N（0〈=N〈=100）

然后输入一个N*N的矩阵 其中第i行第j列的数字k表示从点i到点j需要的花费。

Output

一个数字，最少需要多少花费才能使得整张图任意两点都直接或者间接连通（也就是最小生成树的权）

Sample Input

5
0 41 67 34 0 
41 0 69 24 78 
67 69 0 58 62 
34 24 58 0 64 
0 78 62 64 0

0

2
0 1
1 0

Sample Output

116
0
1

解答

#include <bits/stdc++.h>
using namespace std;
#define MAX 105
#define INF 999999999

struct Tedge{
	int start;
	int end;
	int fee;
};//定义变的结构体 
int fee[MAX][MAX];
bool bv1[MAX],bv2[MAX];//用两个数组判断是否已连通 

void init(){
	memset(bv1,false,sizeof(bv1));//标记已在生成树中的节点 
	memset(bv2,true,sizeof(bv2));//还未包含进生成树的节点 
}

void input(int n){
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			cin>>fee[i][j];
		}
	} 
}

Tedge getEdge(int n){//获取权值最小（花费最少）的边 
	Tedge tRet;
	int minFee=INF;
	int nstart,nend;
	for(int i=1;i<=n;i++){
		if(bv1[i]){
			for(int j=1;j<=n;j++){
				if(bv2[j]){
					if(minFee>fee[i][j]){
						minFee=fee[i][j];
						nstart=i;
						nend=j;
					}
				}
			}
		}
	} 
	tRet.start=nstart;
	tRet.end=nend;
	tRet.fee=minFee;
	return tRet;
}

int Prim(int n){
	int nans=0,nC=1;
	Tedge tA;
	bv1[1]=true;//将第一个节点设置为已经加入生成树中 
	bv2[1]=false;
	for(int i=1;i<n;i++){
		tA=getEdge(n);//找到剩余节点中与生成树连接的边中权值最小的边
		bv1[tA.end]=true;//将该边连接的节点加入到生成树 
		bv2[tA.end]=false;
		nans+=tA.fee;
	}
	return nans;
}

int main(){
	int n,ans=0;
	while(cin>>n){
		init(); 
		input(n);
		if(n!=0){
			ans=Prim(n);
		}else{
			ans=0;
		}
		cout<<ans<<endl;
	}
	return 0;
}