力扣--图论/Prim1584.连接所有点的最小费用

思路分析:

  1. 初始化 :获取点的数量,并创建两个辅助数组 adjvexlowcost,分别用于记录最小生成树的边信息和每个顶点到最小生成树的距离。
  2. Prim算法循环 :在每一次循环中,选择一个未加入最小生成树的顶点 k,使得从已加入最小生成树的顶点到 k 的距离最小。循环 n-1 次,每次选择一个顶点加入最小生成树。
  3. 找出下一个顶点 :遍历所有未加入最小生成树的顶点,选择距离最小的顶点 k,加入最小生成树,并标记顶点 k 已被访问。
  4. 更新最小生成树信息 :更新 lowcost 数组,更新每个顶点到最小生成树的距离。
  5. 计算总成本:计算最小生成树的总成本,即所有边的长度之和,并返回。
cpp 复制代码
class Solution {
public:
    int minCostConnectPoints(vector<vector<int>>& points) {
        int n = points.size(); // 获取点的数量

        // 用于存储最小生成树的边信息
        vector<int> adjvex(n, 0); // 记录最小生成树的顶点的下标
        vector<int> lowcost(n, 0); // 记录从最小生成树中的某顶点到其它顶点的最小权值

        // 初始化lowcost数组,将除了第一个点以外的顶点到第一个点的距离记录下来
        for(int i = 1; i < n; i++)
            lowcost[i] = abs(points[i][0] - points[0][0]) + abs(points[i][1] - points[0][1]);

        // 用于记录顶点是否已被访问过
        vector<bool> visited(n, false);

        // Prim算法的主要循环,构建最小生成树
        for(int t = 0; t < n - 1; t++) {
            int k = 1, min = INT_MAX;
            // 找出lowcost数组中的最小值
            for(int i = 1; i < n; i++) {
                if(lowcost[i] < min && visited[i] == false) {
                    min = lowcost[i];
                    k = i;
                }
            }
            // 标记顶点k已被访问
            visited[k] = true;
            // 将k顶点到其它顶点的权值设为0,表示已加入最小生成树
            lowcost[k] = 0;
            // 更新lowcost数组中的值
            for(int i = 1; i < n; i++) {
                if(i != k) {
                    // 计算顶点i到顶点k的距离
                    int close = abs(points[i][0] - points[k][0]) + abs(points[i][1] - points[k][1]);
                    // 如果顶点i到顶点k的距离小于当前到顶点i的最小权值,则更新相应信息
                    if(lowcost[i] > close) {
                        adjvex[i] = k;
                        lowcost[i] = close;
                    }
                }
            }
        }

        // 计算最小生成树的总成本
        int num = 0;
        for(int i = 0; i < n; i++) {
            num += abs(points[i][0] - points[adjvex[i]][0]) + abs(points[i][1] - points[adjvex[i]][1]);
        }
        // 返回最小生成树的总成本
        return num;
    }
};
相关推荐
心中有国也有家4 分钟前
hixl:昇腾分布式推理的「快递专线」
人工智能·经验分享·笔记·分布式·学习·算法
草莓熊Lotso1 小时前
【Linux系统加餐】从原理到封装:基于建造者模式实现System V信号量工业级C++封装
android·linux·运维·服务器·网络·c++·建造者模式
爱睡懒觉的焦糖玛奇朵7 小时前
【从视频到数据集:焦糖玛奇朵的魔法工具使用说明】
人工智能·python·深度学习·学习·算法·yolo·音视频
Runawayliquor7 小时前
opbase:CANN 所有算子的公共地基
大数据·数据库·人工智能·算法
徐安安ye7 小时前
FlashAttention 为什么对序列长度这么“敏感”?
人工智能·算法
潜创微科技7 小时前
IT6520:USB‑C 转 MIPI 芯片方案 4K@120Hz 高清显示
c语言·开发语言
kyle~8 小时前
机器视觉---熔池相机(穿透强光的视觉感知)
c++·数码相机·计算机视觉·机器人·焊接机器人
宏笋8 小时前
C++ Coroutines(协程) 详解
c++
王老师青少年编程9 小时前
csp信奥赛C++高频考点专项训练之前缀和&差分 --【一维前缀和】:求区间和
c++·前缀和·csp·高频考点·信奥赛·求和区间和
黎阳之光9 小时前
黎阳之光:以视频孪生重构智能监盘,为燃机打造新一代智慧电厂大脑
大数据·人工智能·算法·安全·数字孪生