洛谷 P1546 【最短网络 Agri-Net】

最小生成树，用kruskal法（要用并查集）

实在没力气再把两个算法很详细的说一遍了，如果第一次涉及，可以先看下一本通或者我之前写的题解

基本是裸的最小生成树

读入每条边然后排序，依次枚举

如果两点没有在同一个集合里，就合并，答案加上这条边的边权，边数加一，当边数等于n-1（如果更小不可能把每个点都连接）时退出循环

#include<iostream>
#include<algorithm>
using namespace std;
int n;
struct node{
	int a,b;
	int jl;
}dis[10007];//保存边
int fa[107];//并查集，保存祖先
int ans;//答案
int bs;//边数
int fi(int x){
	if(fa[x]!=x)fa[x]=fi(fa[x]);
	return fa[x];//找祖先
}
void hb(int x,int y){
	fa[x]=y;
	return;//合并
}
bool cmp(node x,node y){
	return x.jl<y.jl;//按边权排序
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++)
	for(int j=1;j<=n;j++){
    		//这里读入要注意下，第x条边可以表示为第（i-1）*n+j条边
		dis[(i-1)*n+j].a=i;
		dis[(i-1)*n+j].b=j;
		cin>>dis[(i-1)*n+j].jl;
		if(dis[(i-1)*n+j].jl==0)dis[i].jl=0x3f3f3f3f;//不然就会排在前面，到时候判断也行
	}
	for(int i=1;i<=n;i++)
	fa[i]=i;//先把祖先改成每个人自己
	sort(dis+1,dis+n*n-n+1,cmp);//排序
	for(int i=1;i<=n*n-n;i++){
		int x=dis[i].a;
		int y=dis[i].b;//为写代码方便
		if(fi(x)!=fi(y)){//不在同一集合
			hb(fi(x),fi(y));//合并
			bs++;//边数加一
			ans+=dis[i].jl;//加上边权
		}
		if(bs==n-1)break;//已全部连接
	}
	cout<<ans<<endl;//输出
	return 0;
}