最小生成树模板

最新推荐文章于 2022-05-14 21:31:35 发布

looooooogn

最新推荐文章于 2022-05-14 21:31:35 发布

阅读量93

点赞数

本文链接：https://blog.csdn.net/qq_40273481/article/details/80739215

版权

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f

 struct Edge
{
    int u,v,w;
}edge[MAXM];

int tot;
void addedge(int u,int v,int w)
{
    edge[tot].u=u;
    edge[tot].v=v;
    edge[tot++].w=w;
}//   注意 加边时是有向边！！！
//    所以如果是无向图求最小生成树的话，
//    addedge( u,v,w ), addedge( v,u,w ) 两次加边 ( 即反向 )即可 

bool cmp(Edge a,Edge b)
{
    return a.w<b.w;
}

int Find(int x)
{
    if(F[x]==-1)return x;
    else
        return F[x]=Find(F[x]);
}

int kruskal(int n)
{
    memset(F,-1,sizeof(F));
    sort(edge,edge+tot,cmp);
    int cnt=0;
    int ans=0;
    for(int i=0;i<tot;i++){
        int u=edge[i].u;
        int v=edge[i].v;
        int w=edge[i].w;
        int t1=Find(u);
        int t2=Find(v);
        if(t1!=t2){
            ans+=w;
            F[t1]=t2;
            cnt++;
        }
        if(cnt==n-1)break;
    }
    if(cnt<n-1)return -1;//   图不连通  
    else return ans;//   最小生成树权值  

}