【并查集】【克鲁斯卡尔】【邻接表】最优布线问题

Description

学校有n台计算机，为了方便数据传输，现要将它们用数据线连接起来。两台计算机被连接是指它们之间有数据线连接。由于计算机所处的位置不同，因此不同的两台计算机的连接费用往往是不同的。
当然，如果将任意两台计算机都用数据线连接，费用将是相当庞大的。为了节省费用，我们采用数据的间接传输手段，即一台计算机可以间接的通过若干台计算机（作为中转）来实现与另一台计算机的连接。
现在由你负责连接这些计算机，你的任务是使任意两台计算机都连通（不管是直接的或间接的）。

Input

第一行为整数n（2<=n<=100），表示计算机的数目。此后的n行，每行n个整数。第x+1行y列的整数表示直接连接第x台计算机和第y台计算机的费用。

Output

一个整数，表示最小的连接费用。

思路

克鲁斯卡尔。默认每一个点是一个独立的并查集。然后寻找所有并查集之间最短的一条路，用这条路连接俩个不同的并查集。将此路径的长度加入ans。总共要用n-1次将n个并查集合并成一个。

#include<cstdio>
#include<algorithm>
using namespace std;
int n,l,t,k,ans,f[101];
struct asdf{    //存路径用
	int x,y,z;
} a[5001];
bool cmd(asdf aa,asdf bb){  //排序用，用路径长度从小到大排序
	return aa.z<bb.z;
}
int find(int d){     //寻找父节点
	if(f[d]==d) return d;
	f[d]=find(f[d]);  //压缩路径
	return f[d];
}
int main(){
	scanf("%d",&n);  //读入
	for(int i=1;i<=n;++i){
		for(int j=1;j<=n;++j){
	  	    scanf("%d",&l);
	  	    if(i<j&&l){  //避免存重复。。。
	  	    	a[++t].x=i;a[t].y=j;a[t].z=l;  //存下这条路连接的俩个点，存下路程
	  	    }
	    }
	    f[i]=i;  //将父节点赋值为自己
	}
	sort(a+1,a+t+1,cmd);  //排序
	ans+=a[1].z;
	f[find(min(a[1].x,a[1].y))]=f[find(max(a[1].x,a[1].y))];  //合并俩个距离最短的并查集
	k=2; //已经处理过一条路了
	for(int i=2;i<n;++i){
		while(find(a[k].x)==find(a[k].y))  //找一条路，连接的俩个点不在同一个并查集（排序后保证它最短） 
		  ++k;
		f[find(min(a[k].x,a[k].y))]=f[find(max(a[k].x,a[k].y))];  //合并俩个并查集。
		ans+=a[k].z;  //加上路程
	}
	printf("%d",ans);
}