2022.4.7 力扣-连接所有点的最小费用-最小生成树练习

目录

题目描述：

样例：

方法一（普利姆算法-prim算法）

---

本文参考：

(5条消息) 数据结构--最小生成树详解_Ouyang_Lianjun的博客-CSDN博客_最小生成树

Prim and Kruskal(并查集） - 连接所有点的最小费用 - 力扣（LeetCode） (leetcode-cn.com)

题目描述：

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

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

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

样例：

方法一（普利姆算法-prim算法）

class Solution {
public:
    int prim(vector<vector<int>>& points, int start)
    {
        int n = points.size();
        vector<vector<int>> graph(n, vector<int>(n));
        //将points数组转化为邻接矩阵
        for (int i = 0; i < n; i++)
        {
            for (int j = i + 1; j < n; j++)
            {
                //曼哈顿距离
                int dis = abs(points[i][0] - points[j][0]) + abs(points[i][1] - points[j][1]);
                graph[i][j] = dis;
                graph[j][i] = dis;
            }
        }
        //记录V中节点到Vnew中的节点的最短距离，如果某个节点i已经加入了Vnew中，则将minlen[i]记为-1
        vector<int> minlen(n);
        //记录某个节点是否已经被访问过，即已经加入了Vnew中
        vector<int> visit(n, 0);

        //先将点start加入Vnew中
        visit[start] = 1;
        minlen[start] = -1;
        //刚刚的minlen没有初始化，这时候将除了点start外，其他点的minlen值设为到start的距离
        for (int i = 0; i < n; i++)
        {
            if (i == start) continue;
            minlen[i] = graph[i][start];
        }
        //ans为最小生成树的权值之和
        int ans = 0;
        //开始prim算法
        //刚才已经加入了一个点start了，只需再循环n - 1次即可
        for (int i = 1; i < n; i++)     //这里i无实际意义，只用于循环
        {
            int cur_minindex = -1;
            int cur_minlen = INT_MAX;
            for (int j = 0; j < n; j++)
            {
                //已经加入过Vnew的节点就不用考虑了
                if (visit[j]) continue;
                if (cur_minlen > minlen[j])
                {
                    cur_minindex = j;
                    cur_minlen = minlen[j];
                }
            }
            //将找到的cur_minindex加入Vnew中
            ans += cur_minlen;
            visit[cur_minindex] = 1;
            minlen[cur_minindex] = -1;
            //由于加了一个新节点到Vnew中，所以更新其他点到Vnew的最小值
            for (int j = 0; j < n; j++)
            {
                if (visit[j]) continue;
                minlen[j] = min(minlen[j], graph[j][cur_minindex]);
            }
        }
        return ans;
    }

    int minCostConnectPoints(vector<vector<int>>& points) {
        return prim(points, 0);
    }
};

复习最小生成树，第一次写代码实现有点磕磕碰碰，prim算法与dijkstra算法在理论和实现上都有相似之处，可以互相比较学习。

时间复杂度：O(n * n)

空间复杂度：O(n * n)

//未完待续