从邻接矩阵中提取出边,然后跑一边kruscal
1 #include<iostream> 2 #include<algorithm> 3 using namespace std; 4 struct node 5 { 6 int x,y,w; 7 }; 8 int cnt=0;//记录边数 9 node a[100*100];//记录边 10 int fa[110];//记录并查集 11 int finds(int x)//并查集找祖先 12 { 13 while(fa[x]!=x) 14 { 15 x=fa[x]; 16 } 17 return x; 18 } 19 bool cmp(node a,node b) 20 { 21 return a.w<b.w; 22 } 23 int main(void) 24 { 25 int n; 26 cin>>n; 27 for(int i=1;i<=n;i++)//提取边 28 { 29 for(int j=1;j<=n;j++) 30 { 31 int t; 32 cin>>t; 33 if(j>i)//如果列大于行 34 { 35 a[cnt].x=i; 36 a[cnt].y=j; 37 a[cnt].w=t; 38 cnt++; 39 } 40 } 41 } 42 sort(a,a+cnt,cmp);//排序 43 //以上处理边 44 for(int i=1;i<=n;i++) 45 { 46 fa[i]=i; 47 }//初始化并查集 48 int res=0; 49 for(int i=0;i<cnt;i++)//找最小生成树 50 { 51 int from=a[i].x; 52 int to=a[i].y; 53 int weight=a[i].w; 54 int f1=finds(from); 55 int f2=finds(to); 56 if(f1!=f2) 57 { 58 res+=weight; 59 fa[f1]=f2;//union 60 } 61 } 62 cout<<res; 63 return 0; 64 }
原文地址:https://www.cnblogs.com/greenofyu/p/12233887.html
时间: 2024-10-13 20:23:16