1349:【例4-10】最优布线问题
时间限制: 1000 ms 内存限制: 65536 KB
提交数: 2371 通过数: 1461
【题目描述】
学校有n台计算机,为了方便数据传输,现要将它们用数据线连接起来。两台计算机被连接是指它们有数据线连接。由于计算机所处的位置不同,因此不同的两台计算机的连接费用往往是不同的。
当然,如果将任意两台计算机都用数据线连接,费用将是相当庞大的。为了节省费用,我们采用数据的间接传输手段,即一台计算机可以间接的通过若干台计算机(作为中转)来实现与另一台计算机的连接。
现在由你负责连接这些计算机,任务是使任意两台计算机都连通(不管是直接的或间接的)。
【输入】
第一行为整数n(2≤n≤100),表示计算机的数目。此后的n行,每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。
【输出】
一个整数,表示最小的连接费用。
【输入样例】
3
0 1 2
1 0 1
2 1 0
【输出样例】
2
【提示】
注:表示连接1和2,2和3,费用为2。
【来源】
No
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=101;
const int M=N*N;
struct node {
int u,v,w;
} e[M];
int i,j,n,m=0,p=0,fa[N],s=0;
bool fangfa (const node &x,const node &y) {
return x.w<y.w;
}
int find (int x) {
if(fa[x]!=x) fa[x]=find(fa[x]);
return fa[x];
}
void kru() {
int k=0;
for(i=1;i<=n;i++)
fa[i]=i;
sort(e+1,e+m+1,fangfa);//初始化
for(i=1;i<=m;i++) {
int ui=e[i].u;int vi=e[i].v;
int u1=find(ui);int v1=find(vi);
if(u1!=v1) {//如果起始点不在一个集合,就进行操作
fa[u1]=v1;//合并节点
k++;
s+=e[i].w;//累加结果
if(k==n-1) break;
}
}
}
int main () {
int c;
scanf("%d",&n);
for(i=1;i<=n;i++)
for(j=1;j<=n;j++) {
scanf("%d",&c);
if(i!=j&&i<j) {//因为是无向图,所以矩阵是对称的,只用保存一半即可
m++;
e[m].u=i;e[m].v=j;e[m].w=c;
}
}
kru();
printf("%d",s);
return 0;
}