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;
}