XYNU1392: 最优布线问题(prim)

1392: 最优布线问题

时间限制: 1 Sec 内存限制: 128 MB
提交: 18 解决: 12
您该题的状态:已完成
[提交][状态][讨论版]
题目描述
学校有n台计算机,为了方便数据传输,现要将它们用数据线连接起来。两台计算机被连接是指它们间有数据线连接。由于计算机所处的位置不同,因此不同的两台计算机的连接费用往往是不同的。

当然,如果将任意两台计算机都用数据线连接,费用将是相当庞大的。为了节省费用,我们采用数据的间接传输手段,即一台计算机可以间接的通过若干台计算机(作为中转)来实现与另一台计算机的连接。

现在由你负责连接这些计算机,任务是使任意两台计算机都连通(不管是直接的或间接的)。
输入
输入文件wire.in,第一行为整数n(2<=n<=100),表示计算机的数目。此后的n行,每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。

输出
输出文件wire.out,一个整数,表示最小的连接费用。
样例输入
3
0 1 2
1 0 1
2 1 0
样例输出
2

#include<iostream>
#include<cstring>
#define max_size 100
#define inf 0x3f3f3f3f
using namespace std;
int map[max_size][max_size];
int dis[max_size];
int N,q,a,b,c;
using namespace std;
int prim(int x){
	memset(dis,0x3f,sizeof(dis));//初始化
	int next = 0,min = 0,sum = 0;
	for(int index=0;index!=-1;index=next){
		sum += min;//获取权重的和
		next = -1;//循环终止的条件,如果没有符合下面两个判断的情况时说明最小生成树建立,此时循环终止
		min = inf;//初始化
		dis[index] = -1;//标记这一点已经被用 
		for(int i = 0; i < x; i++){
			if(map[index][i] < dis[i]){//可以拿到所有和这个点有关系点的距离,包括自身和自身,当然此时为零
				dis[i] = map[index][i];
			}
			if(dis[i]!=-1&& min > dis[i]){
				min = dis[i];//获取到这一点和下一点的最短距离
				 next = i;//拿到下一个点的位置
			}
		} 
	}
	return sum;
}
int main(){
	int n;
	cin>>n;
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++){
			cin>>map[i][j];
		}
	}
	int ans = prim(n);
	cout<<ans<<endl;
	return 0;
}
上一篇:最小生成树-Prim算法


下一篇:最小生成树之Prim算法