题目链接:https://vjudge.net/problem/POJ-1287
最小生成树,点最多50,边却没有限制,按理说应该用Prim+邻接矩阵
也可以先用邻接矩阵保存两点间最小边,再加入边集,再用Kruskal做
这里用了后者
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAXN=50+10;
const int MAXM=3000+10;
int UF[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(UF[x]==-1) return x;
else return UF[x]=find(UF[x]);
}
int Kruskal(int n)
{
memset(UF,-1,sizeof(UF));
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)
{
UF[t1]=t2;
ans+=w;
cnt++;
}
if(cnt==n-1) break;
}
if(cnt<n-1) return -1;
return ans;
}
int g[MAXN][MAXN];
int main()
{
int n,m;
while(~scanf("%d",&n)&&n)
{
int a,b,c;
memset(g,-1,sizeof(g));
scanf("%d",&m);
for(int i=0;i<m;i++)
{
scanf("%d%d%d",&a,&b,&c);
if(g[a][b]==-1||g[a][b]>c) g[a][b]=g[b][a]=c;
}
tol=0;
for(int i=1;i<=n;i++)
{
for(int j=1;j<i;j++)
{
if(g[i][j]!=-1)
addedge(i,j,g[i][j]);
}
}
printf("%d\n",Kruskal(n));
}
return 0;
}