最小生成树 Kruskal算法 Prim算法 洛谷P3366

最小生成树 Kruskal算法 Prim算法 洛谷P3366

相较于Prim算法，我觉得Kruskal算法更优（因为一般情况，题目给你的边数都是正常的，Kruskal算法的时间复杂度为O(ElogE) E为边的数量），Kruskal算法是将边按权值排序之后利用并查集进行加边，直至加入边的数量为n-1时结束，而Prim算法是每一次加入一个点，使用的是邻接表进行加点，其时间复杂度为O(n^2)，n为点的数量。

当然Prim算法经过优化后也是很强大的： Prim算法循环 n - 1，每次都要寻找距离集合Vnew的最小值， 扫描与一个点所连接的所有边。如果使用将一个点所有的边都扫描一遍的算法，则时间复杂度为O(n² + E)。如果我们使用二叉堆来实现查找距离集合Vnew的最小值，则时间复杂度为O(E logV )。如果使用斐波那契堆优化的话，那么时间复杂度将可以近一步优化为O( E + V logV)。

所以，Prim算法适用于点很少，但是边很多的情况，即适用于稠密图的情况，相反Kruskal适用于边很少但点很多的情况。

以洛谷的这个模板题来说，两者均适用

Kruskal算法：

#pragma GCC optimize("O2")
#include <bits/stdc++.h>
using namespace std;
const int maxn=2e5+7;
int n,m,tot=0,k=0;
int fat[maxn];//并查集
struct Node{
    int fr,to,dis;
    bool operator <(const Node x)const{
        return dis<x.dis;
    }
}Edge[maxn];
int fa(int x){
    if(fat[x]!=x)
        return fa(fat[x]);
    return x;
}
void connect(int x,int y){
    fat[fa(y)]=fa(x);
}
int main(){
    scanf("%d %d",&n,&m);
    for(int i=1;i<=m;i++)
        scanf("%d %d %d",&Edge[i].fr,&Edge[i].to,&Edge[i].dis);
    for(int i=1;i<=n;i++)
        fat[i]=i;//并查集初始化
    sort(Edge+1,Edge+1+m);//将边按权值排序
    for(int i=1;i<=m;i++){
        if(k==n-1)    break;//边数达到n-1时就退出
        if(fa(Edge[i].fr) != fa(Edge[i].to)){
            connect(Edge[i].fr,Edge[i].to);//将两个点用并查集连接起来
            tot+=Edge[i].dis;//加上加入边的边权
            k++;//边数++
        }
    }
    printf("%d",tot);//输出最小生成树的权值
return 0;
}

Prim算法(未优化)：Prim算法可以通过二叉堆将时间复杂度优化到O(ElogV)，斐波那契堆更是可以优化到O(E+VlogV),这些版本我之后再补吧。

#include<bits/stdc++.h>
using namespace std;
#define inf 0x7fffffff
#define maxn 5005
#define maxm 200005
struct edge{
    int v,w,next;
}e[maxm<<1];
int head[maxn],dis[maxn],cnt,n,m,tot,now=1,ans;
bool vis[maxn];
inline int read(){
	int x=0,f=1;char c=getchar();
    while(c<'0'||c>'9'){if(c=='-') f=-1;c=getchar();}
    while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
    return x*f;
}
inline void add(int u,int v,int w){
    e[++cnt].v=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt;
}//向前星邻接表
inline void init(){
    n= read(),m= read();
    for(  int i=1,u,v,w;i<=m;++i){
        u= read(),v= read(),w= read();
        add(u,v,w),add(v,u,w);//建立邻接表
    }
}
inline int prim(){
    for(int i=2;i<=n;++i)
        dis[i]=inf;
    for(int i=head[1];i;i=e[i].next)
        dis[e[i].v]=min(dis[e[i].v],e[i].w);//更新所有剩余的点到当前节点的距离
    while(++tot<n){
        int minn=inf;
        vis[now]=1;
        for(int i=1;i<=n;++i)
            if(!vis[i]&&minn>dis[i])
                minn=dis[i],now=i;//首先从起始位置选出一个离我们需要的点集最近的点
        ans+=minn;
        for(int i=head[now];i;i=e[i].next){
            int v=e[i].v;
            if(dis[v]>e[i].w&&!vis[v])
                dis[v]=e[i].w;//更新所有剩余的点到当前节点的距离
        }
    }
return ans;
}
int main(){
    init();
    printf("%d",prim());
return 0;
}