克鲁斯卡尔算法——求加权连通图的最小生成树的kruskal套路

#include
#include
#include 
using namespace std;
//MAX取具体边的值
#define MAX 5005
//n,m为点与边的数量
int n,m;
//结构体数据类型的边变量
struct Edge
{
    //边连接的2点与权重
    int x,y,w;
}e[MAX];
//记录点的树根
int k[MAX];
//排序用函数,此处为按权值从小到大排序
bool cmp(Edge a,Edge b)
{
    return a.w//递归省时间费空间,用于找树根
int find(int x)
{
    if(k[x] != x)k[x] = find(k[x]);
    return k[x];
}
//另一种神奇写法
/*
int find(int x)
{
    return father[x]==x?x:father[x]=find(father[x]);
}
*/
//循环写法
/*
int find(int p)
{
    int q=p;
    while(node[q]!=q)
        q=node[q];
    return q;
}
*/
//判断某边是否能加入,不能则返回0,能则返回权值
int join(Edge p)
{
    int i=find(p.x),j=find(p.y);
    //若树根相等则成环需排除
    if(i!=j)
    {
        k[i]=j;
        return p.w;
    }
    return 0;
}
void kruskal()
{
    int i;
    //初始化
    for(i=1;i<=n;i++)
    {
        k[i]=i;
    }
    int sum=0;
    //排序,还记得之前的cmp吗(若之前赋值有e[0]开始,则省去+1)
    sort(e+1,e+1+m,cmp);
    //核心,边权由小到大依次判断增加(此处m为边的数量)
    for(i=1;i<=m;i++)
        sum+=join(e[i]);
    //输出结果
    printf("%d\n",sum);
}

你可能感兴趣的:(模板)