【Kruskal算法】 -- poj1287

题目来源: poj 1287

题意：

连通一个图，找最小权值和，有重边

要点：

因为有重边，所以Prim算法还要取最小的边加入，而Kruskal算法可以直接忽略这个问题，因为一开始要排序，大的自动舍掉了。

#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 100;
const int maxm = 5100;
struct edge
{
    int u, v;
    int w;
    edge() {}
    edge(int _u, int _v, int _w) : u(_u), v(_v), w(_w) {}
} E[maxm];

bool cmp(edge e1, edge e2)
{
    return e1.w < e2.w;
}

//并查集部分
int father[maxn];

int find(int x)
{
    if (x == father[x])
    {
        return x;
    }
    else
    {
        int F = find(father[x]);
        father[x] = F;
        return F;
    }
}

//kruskal部分
int kruskal(int n, int m)
{
    int ans = 0;                 //所求边权之和
    int num_edge = 0;            //当前生成树包含的边的数目
    for (int i = 1; i <= n; i++) //并查集初始化，顶点编号范围是[1,n]!!!
    {
        father[i] = i;
    }
    sort(E, E + m, cmp);
    for (int i = 0; i < m; i++)
    {
        int faU = find(E[i].u);
        int faV = find(E[i].v);
        if (faU != faV)
        {
            father[faU] = faV; //合并集合，即把测试边加入最小生成树中
            num_edge++;
            ans += E[i].w;
            if (num_edge == n - 1)
            {
                break;
            }
        }
    }
    if (num_edge != n - 1)
    {
        return 0; //无法连通时返回0
    }
    else
    {
        return ans;
    }
}

int main()
{
    int n, m;
    while (cin >> n && n != 0)
    {
        cin >> m;
        for (int i = 0; i < m; i++)
        {
            int u, v, w;
            cin >> u >> v >> w;
            E[i] = edge(u, v, w);
        }
        cout << kruskal(n, m) << endl;
    }
    return 0;
}