图论之克鲁斯卡尔（kruskal）算法

最小生成树之克鲁斯卡尔算法

例题

给你一个points 数组，表示 2D 平面上的一些点，其中 points[i] = [xi, yi] 。

连接点 [xi, yi] 和点 [xj, yj] 的费用为它们之间的 曼哈顿距离 ：|xi - xj| + |yi - yj| ，其中 |val| 表示 val 的绝对值。

请你返回将所有点连接的最小总费用。只有任意两点之间 有且仅有 一条简单路径时，才认为所有点都已连接。

示例 1：

输入：points = [[0,0],[2,2],[3,10],[5,2],[7,0]]
输出：20
解释：

我们可以按照上图所示连接所有点得到最小总费用，总费用为 20 。
注意到任意两个点之间只有唯一一条路径互相到达。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/min-cost-to-connect-all-points
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

典型的最小生成树问题，这里使用克鲁斯卡尔算法解题，克鲁斯卡尔的核心就是并查集。

代码

class Solution {
    vector<int> parent;
    vector<int> size;
public:
    int find(int x){
        while (x != parent[x]){
            parent[x] = parent[parent[x]];
            x = parent[x];
        }
        return x;
    }
    void Union(int p,int q){
        int rootP = find(p);
        int rootQ = find(q);
        if (rootP == rootQ) return ;
        if (size[rootP] > size[rootQ]){
            parent[rootQ] = parent[rootP];
            size[rootP] += size[rootQ];
        }
        else {
            parent[rootP] = rootQ;
            size[rootQ] += size[rootP];
        }
        return ;
    }
    bool Connected(int p,int q){
        int rootP = find(p);
        int rootQ = find(q);
        return rootP == rootQ;
    }
    // 核心代码段
    int minCostConnectPoints(vector<vector<int>>& points) {
        int n = points.size();
        vector<vector<int>> edges;
        for (int i = 0;i < points.size();++i){
            for (int j = i + 1;j < points.size();++j){
                int w = abs(points[i][0] - points[j][0]) + abs(points[i][1] - points[j][1]);
                edges.push_back({i,j,w});
            }
        }
        sort(edges.begin(),edges.end(),[&](vector<int>& a,vector<int>& b){
            return a[2] < b[2];
        });
        // for (int i = 0;i < edges.size();++i) cout << edges[i][0] << "," << edges[i][1] << "," << edges[i][2] << endl;
        parent = vector<int>(n,0);
        iota(parent.begin(),parent.end(),0);
        size = vector<int>(n,1);
        int ans = 0;
        for (int i = 0;i < edges.size();++i){
            if (Connected(edges[i][0],edges[i][1])) continue;
            Union(edges[i][0],edges[i][1]);
            ans += edges[i][2];
        }
        return ans;
    }
};