ch2(搜索与图论)——最小生成树（Prim算法、Kruskal算法）

1.自然语言描述
Prim 算法是利用贪心思想。每次选距离当前生成树最近的点（注意是到生成树的距离，即该点到邻接的属于生成树的点的距离最小）直至所有的点被选完。
Kruskal 算法是将所有的边按权值非递减排序，每次从中挑选最小的边（注意选择的边不能在图中的生成树上构成环）直至选择够(|V|-1)条边。
2.代码描述
题目：Acwing.858 Prim算法求最小生成树题目链接

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int MAXN=510;

int n,m,dist[MAXN],g[MAXN][MAXN],idx,sum;
bool st[MAXN];

int prim()
{
	//Prim算法代码和dijkstra十分相似，因为都是以贪心策略遍历邻接点的做法
    memset(dist,0x3f,sizeof(dist));
    //注意dist是每个点到生成树的最短距离，刚开始生成树里没有任何点
    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||dist[t]>dist[j]))
                t=j;
        
        if(i&&dist[t]==0x3f3f3f3f)//不是连通图，无最小生成树
            return -1;
        if(i)//i=0时找到的是第一个扩展点，生成树中只有一个点，所以i=0时res仍为0
            res+=dist[t];
            
        st[t]=true;
            
        for(int j=1;j<=n;j++)
            dist[j]=min(dist[j],g[t][j]);//更新所有点的dist状态
    }
    return res;
}

int main(void)
{
    cin>>n>>m;
    
    memset(g,0x3f,sizeof(g));
    
    for(int i=0;i<m;i++){
        int a,b,c;
        cin>>a>>b>>c;
        g[a][b]=g[b][a]=min(g[a][b],c);//注意是无向图
    }
    
    int ans=prim();
    if(ans==-1)
        cout<<"impossible"<<endl;
    else
        cout<<ans<<endl;
    
    return 0;
}

题目：Acwing.859 Kruskal 算法求最小生成树题目链接

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
const int MAXN=2e5+10;

int n,m,p[MAXN];
struct Edge{
    int a,b,z;
    
    bool operator <(const Edge &A){
        return z<A.z;
    }
}w[MAXN];

int find(int x)
{
    if(p[x]!=x)
        return p[x]=find(p[x]);
    return p[x];
}

int main(void)
{
    cin>>n>>m;
    
    for(int i=1;i<=n;i++)
        p[i]=i;
    
    for(int i=0;i<m;i++)
        cin>>w[i].a>>w[i].b>>w[i].z;
        
    sort(w,w+m);
    
    //利用并查集操作，每条边的两个点合并到一个集合中，不是一个集合的才合并（避免形成环）
    int ans=0,cnt=0;//kruskal
    for(int i=0;i<m;i++){
        int a=w[i].a,b=w[i].b,z=w[i].z;
        int pa=find(a),pb=find(b);
        if(pa!=pb){
            p[pa]=pb;
            ans+=z;
            cnt++;
        }
    }
    
    if(cnt<n-1)//选边不超过n-1则未形成生成树
        cout<<"impossible"<<endl;
    else
        cout<<ans<<endl;
    
    return 0;
}