苗条的生成树（UVA1395）

最小生成树----Kruskal算法

模板

//伪代码
/*
把所有边按照从小到大排序，记第i小的边为e[i]
初始化MST为空
初始化联通分量，让每个点自成一个连通分量，即自己的根为自己
for(int i = 0; i < n; i++)
    if(e[i].u和e[i].v不再同一连通分量）
    {
        把编e[i]加入到MST
        合并e[i].u和e[i].v所在的连通分量
    }
*/
//模板
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1000;
int p[maxn],r[maxn];
struct spe
{
    int u,v,w;
}e[maxn];

bool cmp(spe a,spe b)
{
    return a.w < b.w;
}
int find(int x)
{
    return p[x] == x? x: p[x] = find(p[x]);
}

//结构体式
int kruskal()
{
    int ans = 0; //求边权值和
    for(int i = 0; i < n; i++)
        p[i] = i;
    sort(e,e+m,cmp);
    for(int i = 0; i < m; i++)
    {
        int x = find(e[i].u),y = find(e[i].v);
        if(x != y)
        {
            ans += e[i].w;
            p[x] = y;
        }
    }
    return ans;
}

//紫书上模板
bool cmp(int i,int j)
{
    return w[i]<w[j];
}
int find(int x)；//同上
int kruskal()
{
    int ans = 0;
    for(int i = 0; i < n; i++)
        p[i] = i; //每个结点的父节点都为自己
    for(int i = 0; i < m; i++)
        r[i] = i; //存储边权值的序号
    sort(r,r+m,cmp);
    for(int i = 0; i < m; i++)
    {
        int e = e[i];
        int x = find(u[e]), y = find(v[e]);
        if(x != y) //不属于同一连通块
        {
            ans += w[e];
            p[x] = y;
        }
    }
    return ans;
}

题目链接：苗条的生成树（UVA1395） 由此进入

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 5000;
int p[maxn],n,m;;
struct spe
{
    int u,v,w;
}e[maxn];
bool cmp(spe a,spe b)
{
    return a.w < b.w;
}
int find(int x)
{
    return p[x] == x? x : p[x] = find(p[x]);
}
int kruskal(int a)
{
    for(int i = 1; i <= n; i++)
        p[i] = i;
    int cnt = 0;
    for(int i = a; i < m; i++)
    {
        int x = find(e[i].u),y = find(e[i].v);
        if(x != y)
        {
            cnt++;
            if(cnt == n-1)
                return e[i].w-e[a].w;
            p[x] = y;
        }
    }
    return INF;
}
int solve()
{
    if(m < n-1||kruskal(0) == INF ) return -1;
    int ans = INF;
    sort(e,e+m,cmp);
    for(int i = 0; i < m; i++)
    {
        int total = kruskal(i);
        if(total == INF)
            continue;
        ans = min(ans,total);
    }
    return ans;
}
int main()
{
    while(~scanf("%d%d",&n,&m)&& (n!=0||m!=0))
    {
        for(int i = 0; i < m; i++)
        {
            scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
        }
        int temp = solve();
        cout<<temp<<endl;
    }
    return 0;
}