E - 数据结构实验之图论九：最小生成树

prim算法

#include <bits/stdc++.h>
#define endl '\n'
#define INF 0x3f3f3f3f
using namespace std;
const int N=105;
int g[N][N];
int dist[N];
bool vis[N];
int n,m;
void init()
{
    for(int i=0;i<=n;i++)
    {
        vis[i]=false;
        dist[i]=INF;
        for(int j=0;j<=n;j++)
        {
            if(i==j)g[i][j]=0;
            else g[i][j]=INF;
        }
    }
}
int prim()
{
    int res=0;
    for(int i=0;i<n;i++)
    {
        int t=-1;//t存储离集合距离最小的点
        for(int j=1;j<=n;j++)//j遍历所有点
        {
            if(!vis[j]&&(t==-1||dist[j]<dist[t]))//以此点没有在集合中为前提
                t=j;
        }
        if(i)res+=dist[t];//如果不是第一个点则加边权
        for(int j=1;j<=n;j++)
        {
            if(!vis[j])dist[j]=min(dist[j],g[t][j]);//更新所有没在集合中的点离集合的距离
        }
        vis[t]=true;
    }
    return res;
}
int main()
{
    while(cin>>n>>m)
    {
        init();
        for(int i=0;i<m;i++)
        {
            int u,v,w;
            cin>>u>>v>>w;
            g[u][v]=g[v][u]=min(g[u][v],w);
        }
        int t=prim();
        cout<<t<<endl;
    }
    return 0;
}

kruskal算法

#include <bits/stdc++.h>
#define endl '\n'
#define INF 0x3f3f3f3f
using namespace std;
const int N=110;
struct Edge
{
    int a,b,w;
    bool operator<(const Edge &other)const
    {
        return w<other.w;
    }
}edge[N*N];
int p[N];//并查集思想，储存每个点的祖宗节点
int n,m,res;
void init()
{
    res=0;
    for(int i=1;i<=n;i++)
        p[i]=i;
}
int find(int x)//带有路径压缩优化的并查集find操作
{
    if(p[x]!=x)p[x]=find(p[x]);
    return p[x];
}
void kruskal()
{
    sort(edge,edge+m);
    for(int i=0;i<m;i++)//从小到大遍历每一条边
    {
        int a=edge[i].a,b=edge[i].b,w=edge[i].w;
        a=find(a),b=find(b);
        if(a!=b)//判断此边的两个连接点是否在一个集合中
        {
            p[a]=b;//将边的两个连接点加入集合中
            res+=w;
        }
    }
}
int main()
{
    while(cin>>n>>m)
    {
        init();
        for(int i=0;i<m;i++)
        {
            int a,b,w;
            cin>>a>>b>>w;
            edge[i]={a,b,w};
        }
        kruskal();
        cout<<res<<endl;
    }
}