最小生成树模板prim和kruskal

题目链接

Prim算法：加点法。两个集合S和T，集合S是已经加入的集合，T是待加入的点的集合，每次在集合T中找距离集合S中的所有点距离最短(dis存储)的点，加入集合S，并用该点更新剩余点距离集合S点最近距离dis。

#include<iostream>
#include<algorithm>
#define MAXN 1010
#define inf 8989898
using namespace std;

int graph[MAXN][MAXN];
int dis[MAXN],vis[MAXN];

int main(){
    int n,m;//1-n
    cin>>n>>m;
    //初始化
    fill(graph[0],graph[0]+MAXN*MAXN,inf);
    
    fill(dis,dis+MAXN,inf);
    
    for(int i=0;i<m;i++){
       int a,b,c;
       cin>>a>>b>>c;
       graph[a][b]=graph[b][a]=c;
    }
    int u=1;
    dis[u]=0;
    long long sum=0;
    for(int i=1;i<=n;i++){
        int minDis=inf,v=-1;
        for(int j=1;j<=n;j++){
            if(!vis[j]&&minDis>dis[j]){
                v=j;
                minDis=dis[j];
            }
        }
        vis[v]=1;
        sum+=dis[v];
        for(int j=1;j<=n;j++){
            if(!vis[j]&&dis[j]>graph[v][j]){
                dis[j]=graph[v][j];
            }
        }
    }
    cout<<sum;
    return 0;
}

---

Kruskal算法：加边法。利用并查集和排序。权重从小达到遍历，每次添加一条边（两端不连通），添加n-1次。

#include<iostream>
#include<algorithm>
#define MAXN 100010
using namespace std;
int p[MAXN];
void init(){
    for(int i=0;i<MAXN;i++)p[i]=i;
}
int find(int a){
    if(a!=p[a])p[a]=find(p[a]);
    return p[a];
}
void merge(int a,int b){
    int pa=find(a),pb=find(b);
    if(pa!=pb)p[pb]=pa;
}
struct Edge{
    int from,to,length;
};
struct Edge edge[MAXN];

bool cmp(Edge&a,Edge &b){
    return a.length<b.length;
}
int main(){
    init();
    int n,m;//1-n
    cin>>n>>m;
    for(int i=0;i<m;i++){
       int a,b,c;
       cin>>edge[i].from>>edge[i].to>>edge[i].length;
    }
    sort(edge,edge+m,cmp);
    long long sum=0;
    int ct=0;
    for(int i=0;i<m;i++){
        if(find(edge[i].from)!=find(edge[i].to)){
            merge(edge[i].from,edge[i].to);
            sum+=edge[i].length;
            ct++;
        }
        if(ct==n-1)break;
    }
    cout<<sum;
    return 0;
}

意外收获：

以前没有注意fill和memset函数，这里总结一下。

fill(arr_start,arr_end,value)//数组中元素填充value，头文件是algorithm

memset(arr,value,sizeof(arr))//按照每个字节赋值arr,头文件是string.h

比较：由于memset按字节赋值的特性  memset事实上比 fill 快很多 ，节省了时间