最小生成树

int n; //顶点数
int m; //边数

/*Prim算法*/
int vis[maxn];
int pre[maxn];
int dist[maxn];
int G[maxn][maxn];
void Prim()
{
//边初始化为无穷大
for(int i=0; i<maxn; i++)
for(int j=i; j<maxn; j++)
G[i][j] = G[j][i] = INF;
for(int i=0; i<m; i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
G[u][v] = G[v][u] = w;
}
memset(vis,0,sizeof(vis));
vis[1] = 1;
//最小距离为到出发点的距离
for(int i=1; i<=n; i++)
{
dist[i] = G[1][i];
pre[i] = 1;
}
for(int i=1; i<n; i++)
{
//找到一个距离最小的新的顶点
int u=-1;
for(int j=1; j<=n; j++)
if(!vis[j])
if(u==-1 || dist[u]>dist[j])
u=j;
printf("%d %d %d\n",pre[u],u,G[pre[u]][u]);
vis[u] = 1;
//最小距离更新
for(int j=1; j<=n; j++)
{
if(!vis[j])
{
if(dist[j]>G[u][j])
{
dist[j] = G[u][j];
pre[j] = u;
}
}
}
}
}

/*Kruskal算法(并查集)*/
typedef struct Edge
{
int u;
int v;
int w;
}Edge;
Edge e[maxn];
int fa[maxn];
bool cmp(const Edge &a,const Edge &b)
{
return a.w < b.w;
}
int find_fa(int x)
{
if(fa[x] != x)
fa[x] = find_fa(fa[x]);
return fa[x];
}
void Kruskal()
{
for(int i=0; i<m; i++)
scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
for(int i=0; i<maxn; i++)
fa[i] = i;
sort(e,e+m,cmp);
for(int i=0; i<m; i++)
{
int u = e[i].u;
int v = e[i].v;
int w = e[i].w;
int fa_u = find_fa(u);
int fa_v = find_fa(v);
if(fa_u != fa_v)
{
printf("%d %d %d\n",u,v,w);
fa[fa_v] = fa_u;
}
}
}