POJ 1258 Agri-Net

题意：

给你n个点n*n条边，要你求最小生成树。

思路：

最小生成树Kruskal算法。

#include<cstdio>
#include<algorithm>
using namespace std;

const int MAX=105;
int n,cnt,fa[MAX];

struct Edge{
int u,v,w;
}edge[MAX*MAX];

bool cmp(Edge a,Edge b){
return a.w<b.w;
}

void init(){
for(int i=1;i<=n;i++){
fa[i]=i;
}
}

int Find(int x){
while(x!=fa[x]){
x=fa[x];
}
return x;
}

int Kruskal(){
init();
sort(edge,edge+cnt,cmp);
int ans=0,num=0;
for(int i=0;i<cnt;i++){
int u=Find(edge[i].u);
int v=Find(edge[i].v);
if(u!=v){
ans+=edge[i].w;
fa[u]=v;
num++;
}
if(num>=n-1) break;
}
return ans;
}

int main(){
while(~scanf("%d",&n)){
int w;
cnt=0;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
scanf("%d",&w);
edge[cnt].u=i,edge[cnt].v=j,edge[cnt].w=w;
cnt++;
}
}
int ans=Kruskal();
printf("%d\n",ans);
}
return 0;
}