最小生成树

朴素版prim

---

稠密图

在这里 dis[ ]代表点到集合的距离 st[ ]代表该点有没有加入集合

注意区分prim和dijkstra的区别：prim里dis[ ]代表点到集合的距离 dijkstra代表点到源点的距离

流程：

1.初始化所有点到集合的距离为正无穷

2.一开始所有点距离集合的距离都是正无穷 所以随便找到一个点作为起点并加入集合

3.用这个起点更新其他点到集合（目前该集合只有这个起点）的距离

4.继续找目前距集合最近的点 如果找到的这个点还是距集合正无穷 说明当前的集合和谁都不连通 因为最小生成树要包含所有顶点 所以这个图没有最小生成树 如果找到一个距离最小的点 就把他加入集合 并且更新最小生成树res的值

5.用这个距离最小的点再去更新其他的距离 重复4步骤

这里虽然我一直说的是把点加入集合 但是在我的理解看来 其实除了第一次 是一次次地把一条条边加入了集合

代码：Prim算法求最小生成树 - AcWing题库

#include<iostream>
#include<cstring>
using namespace std;
const int N=510,INF=0x3f3f3f3f;
int g[N][N];
bool st[N];
int dis[N];
int n,m;

int prim()
{
    memset(dis,0x3f,sizeof dis);
    int res=0;//记录最小生成树的大小
    for(int i=0;i<n;++i)
    {
        int t=-1;
        for(int j=1;j<=n;++j)//找到距集合距离最小的点
        {
            if(!st[j]&&(t==-1||dis[t]>dis[j]))
                t=j;
            if(i==0)//如果是第一次找点 随便找到一个点作为起点就好了
                break;
        }
        if(i&&dis[t]==INF)//如果不是第一次循环并且距集合距离正无穷 说明其他点与起点都不连通
            return INF;
        if(i)
            res+=dis[t];
        st[t]=true;//该点收录集合
        for(int j=1;j<=n;++j)//这个对其他点的更新一定要放在后面 防止自环改变了dis[t]的值
            dis[j]=min(dis[j],g[t][j]);
    }
    return res;
}

int main()
{
    int a,b,c;
    cin>>n>>m;
    memset(g,0x3f,sizeof g);
    while(m--)
    {
        cin>>a>>b>>c;
        g[a][b]=g[b][a]=min(g[a][b],c);//防止重边
    }
    int t=prim();
    if(t==INF)
        cout<<"impossible"<<endl;
    else
        cout<<t<<endl;
}

Kruskal算法

---

不需要用数据结构存 用个结构体就可以了

流程：

1.将所有边按权重从小到大排序

2.枚举每条边a,b,c 如果a,b不连通 就将这条边加入集合中（简单的并查集应用）

代码：

#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e5+10,M=2*N;
int p[N];
int findd(int x)
{
    if(x!=p[x])
        p[x]=findd(p[x]);
    return p[x];
}
struct edge
{
    int a,b,w;
}edge[M];
int cmp(struct edge x,struct edge y)
{
    return x.w<y.w;
}
int main()
{
    int n,m;
    int a,b,w;
    cin>>n>>m;
    for(int i=0;i<m;++i)
    {
        cin>>a>>b>>w;
        edge[i]={a,b,w};
    }
    for(int i=1;i<=n;++i)
        p[i]=i;//初始化并查集
    sort(edge,edge+m,cmp);//按权值从小到大排序
    int res=0,num=0;
    for(int i=0;i<m;++i)
    {
        int x=findd(edge[i].a);
        int y=findd(edge[i].b);
        if(x!=y)//如果不连通
        {
            p[x]=y;//连通到一起
            num++;//边数加一
            res+=edge[i].w;//把该边加入集合
        }
    }
    if(num<n-1)//n条顶点应该有n-1条边 如果没有说明没有全部连通
        puts("impossible");
    else
        cout<<res<<endl;
}

//c++prime 去学重载运算符