每周一算法：Kruskal算法求最小生成树

题目名称

[局域网]

题目描述

某个局域网内有 $n$ 台计算机和 $k$ 条双向网线，计算机的编号是 $1\sim n$。由于搭建局域网时工作人员的疏忽，现在局域网内的连接形成了回路，我们知道如果局域网形成回路那么数据将不停的在回路内传输，造成网络卡的现象。注意：

• 对于某一个连接，虽然它是双向的，但我们不将其当做回路。本题中所描述的回路至少要包含两条不同的连接。
• 两台计算机之间最多只会存在一条连接。
• 不存在一条连接，它所连接的两端是同一台计算机。

因为连接计算机的网线本身不同，所以有一些连线不是很畅通，我们用 $f(i,j)$ 表示 $i,j$ 之间连接的畅通程度，$f(i,j)$ 值越小表示 $i,j$ 之间连接越通畅。

现在我们需要解决回路问题，我们将除去一些连线，使得网络中没有回路且不影响连通性（即如果之前某两个点是连通的，去完之后也必须是连通的），并且被除去网线的 $\Sigma f(i,j)$ 最大，请求出这个最大值。

输入格式

第一行两个正整数 $n,k$。

接下来的 $k$ 行每行三个正整数 $i,j,m$ 表示 $i,j$ 两台计算机之间有网线联通，通畅程度为 $m$。

输出格式

一个正整数，表示被除去网线的 $\Sigma f(i,j)$ 的最大值。

样例 #1

样例输入 #1

5 5
1 2 8
1 3 1
1 5 3
2 4 5
3 4 2

样例输出 #1

8

提示

【数据范围】
$1\le n\le 100$,
$0\le k\le 200$,
$1\le f(i,j)\le 1000$

算法思想

根据题目描述，给定一个有 $n$ 台计算机和 $k$ 条双向网线组成的局域网，在不影响连通性的情况下，除去网线产生 $\Sigma f(i,j)$ 的最大值。

为了保证在不影响连通性的情况下去掉网线，可以考虑使用图的生成树算法，由于$f(i,j)$ 值越小表示 $i,j$ 之间连接越通畅，那么可以使用最小生成树算法。最小生成树的算法很多，常用的有两种「Prim」算法和「Kruskal」算法。

Kruskal 算法是一种常见并且好写的最小生成树算法，该算法的基本思想是从小到大加入边。具体来说，维护一个森林，查询两个结点是否在同一棵树中，连接两棵树。这里可以将森林抽象为一堆集合，查询两个元素是否属于同一集合，合并两个集合。其中，查询两点是否连通和连接两点可以使用并查集维护。

时间复杂度

如果使用 $O(m\log m)$ 的排序算法，并且使用并查集进行查询和合并，就可以将时间复杂度做到 $O(m\log m)$ 。

代码实现

#include <bits/stdc++.h>
using namespace std;
const int N = 100010, M = 2 * 10;
struct Edge
{
    int a, b, w;
    bool operator < (const Edge &t)
    {
        return w < t.w;
    }
}e[N];
int n, m;
int p[N]; //并查集记录父节点
int find(int x) //并查集查询根节点
{
    if(x != p[x]) p[x] = find(p[x]); //路径压缩
    return p[x];
}
int main()
{
    cin >> n >> m;

    for(int i = 0; i < m; i++)
    {
        int a, b, w;
        cin >> a >> b >> w;
        e[i] = {a, b, w};
    }

    for(int i = 1; i <= n; i ++) p[i] = i; //并查集初始化

    sort(e, e + m); //按边权从小到大排序

    int res = 0; //统计去掉的边权最大值
    for(int i = 0; i < m; i ++)
    {
        int a = e[i].a, b = e[i].b, w = e[i].w;
        int pa = find(a), pb = find(b);
        //不连通，使用边权最小的边将其连通
        if(pa != pb) p[pa] = b;
        //连通，说明当前边权值w不是最小，要删除掉当前边，累加去掉的边权
        else res += w;
    }
    cout << res << endl;
    return 0;
}