最小生成树(prim&kruskal)模板

prim:

O(v^2),适用于稠密图
const int N=1000;
const int INF=0x3f3f3f3f;
int a[N][N],p[N],low[N];///邻接矩阵 
int prim(int n)
{
    int i,j,ans=0,poi;
    memset(p,0,sizeof(p));
    p[1]=1;
    for(i=1;i<=n;i++)
        low[i]=a[1][i];
    for(i=1;i<n;i++)           n-1次操作
    {
        int mi=INF;
        for(j=1;j<=n;j++)
        {
            if(!p[j]&&mi>low[j])
            {
                mi=low[j];
                poi=j;
            }
        }
        p[poi]=1;
        ans+=mi;
        for(j=1;j<=n;j++)
            if(!p[j])
                low[j]=min(low[j],a[poi][j]);
    }
    return ans;
}
kruskal:将边按权值从小到大排序,枚举每一条边,若该边的两端不在同一集合中,合并集合。直到找到n-1条边(n为点数)。判断是否在一个集合也可以用并查集做

O(ElogE)  适用于稀疏图
const int N=1000;
const int M=N*N;
int p[N];
struct edge
{
    int u,v,w;
}e[M];
int cmp(edge a,edge b)
{
    return a.w<b.w;
}
int kruskal(int n,int m)
{
    sort(e+1,e+m+1,cmp);           下标从1开始
    int i,j,k,ans=0,num=0;
    for(i=1;i<=n;i++)
        p[i]=i;
    for(i=1;i<=m&&num<=n-1;i++)
    {
        判断两端点所在集合
        for(j=e[i].u;p[j]!=j;j=p[j])
            p[j]=p[p[j]];
        for(k=e[i].v;p[k]!=k;k=p[k])
            p[k]=p[p[k]];
        if(j!=k)
        {
            p[j]=k;
            num++;
            ans+=e[i].w;
        }
    }
    return ans;
}



  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值