prime算法-构造最小生成树

1.算法原理

将所有节点分成两个集合V,U。
在带权连通图中V是包含所有顶点的集合，U是已经在最小生成树中的节点；

（1）初始时，从图中任意某一顶点v开始，此时集合U={v}（以v到其他顶点的所有边为侯选边）；
（2） 在所有u∈U,w∈V-U的边(u,w)∈E中找到一条权值最小的边，将(u,w)这条边加入到已找到边的集合，并且将点w加入到集合U中，
（3）重复上一操作，当U=V时，就找到了这颗最小生成树

总而言之，每次查找U中的点与V-U中点路径权值最小的边，再将这条边连接的V-U中的点加入到最小生成树，一直循环，直到所有的点被加入到最小生成树中，所生成的集合U就是最小生成树。

实现

设立两个数据结构

vector<bool>visited; 表示该点是否被加入到最小生成树中，如果在值为1,不在值为0
vector<int>min_cost; 未在最小生成树中的点到最小生成树中的节点的最小权值

代码参见Leetcode 1584https://leetcode-cn.com/problems/min-cost-to-connect-all-points/

class Solution {
public:
    int minCostConnectPoints(vector<vector<int>>& points) {
        int n=points.size();
        int count=points.size();
        int result=0;
        vector<bool>visited(n,0);
        vector<int>min_cost(n,0);
        //先把0加入
        visited[0]=1;
        min_cost[0]=-1;//为-1的时候表示节点已经加入
        count-=1;
        for(int i=1;i<n;i++){
            min_cost[i]=abs(points[i][0]-points[0][0])+abs(points[i][1]-points[0][1]);
        }
        
        while(count>0){
            int min=INT_MAX;
            int index=-1;
            //找到最小值min_cost并记录下标
            for(int i=1;i<n;i++){
                if(visited[i]==0&&min_cost[i]<min){
                    index=i;
                    min=min_cost[i];
                }
            }
            result+=min;
            //将min加入
            visited[index]=1;
            //更新min_cost
            for(int i=0;i<n;i++){
            	//每次查找权值最小的边U->U-V
                if(visited[i]==0&&(abs(points[i][0]-points[index][0])+abs(points[i][1]-points[index][1]))<min_cost[i]){
                    min_cost[i]=(abs(points[i][0]-points[index][0])+abs(points[i][1]-points[index][1]));
                }
            }
            count--;
        }
        return result;
    }
};