最小生成树,用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;
}