/*
Kruskal算法求MST
*/
#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
#include<fstream>
using namespace std;
const int MAXN=505;//最大點數
const int MAXM=250005;//最大邊數
int F[MAXN];//並查集使用
struct Edge
{
int u,v,w;
}edge[MAXM];//儲存邊的信息,包括起點/終點/權值
int tol;//邊數,加邊前賦值爲0
void addedge(int u,int v,int w)
{
edge[tol].u=u;
edge[tol].v=v;
edge[tol++].w=w;
}
bool cmp(Edge a,Edge b)//排序函數,邊按照權值從小到大排序
{
return a.w<b.w;
}
int Find(int x)
{
if(F[x]==-1)
return x;
else
return F[x]=Find(F[x]);
}
int Kruskal(int n)//傳入點數,返回最小生成樹的權值,如果不連通返回-1
{
memset(F,-1,sizeof(F));
sort(edge,edge+tol,cmp);
int cnt=0;//計算加入的邊數
int ans=0;
for(int i=0;i<tol;i++)
{
int u=edge[i].u;
int v=edge[i].v;
int w=edge[i].w;
int t1=Find(u);
int t2=Find(v);
if(t1!=t2)
{
ans+=w;
F[t1]=t2;
cnt++;
}
if(cnt==n-1)
break;
}
if(cnt<n-1)
return -1;//不連通
else
return ans;
}
int main()
{
//freopen("in.txt","r",stdin);
int T;
cin>>T;
int n;
int c;
while(T--)
{
cin>>n;
tol=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
cin>>c;
addedge(i,j,c);
}
}
cout<<Kruskal(n)<<endl;
}
return 0;
}
Kruskal模板求最小生成樹
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.