次小生成树模板(kruskal)

123 篇文章 0 订阅
46 篇文章 0 订阅

kruskal版的次小生成树。


struct data
{
    int a,b,w;
    bool vis;///初始化0
} p[20010];
vector<int>g[110];
int f[110],len[110][110];
const int oo=1e9;
bool cmp(data a,data b)
{
    if(a.w!=b.w)
        return a.w<b.w;
    if(a.a!=b.a)
        return a.a<b.a;
    return a.b<b.b;
}
int fa(int x)///并查集
{
    if(x!=f[x])
        return f[x]=fa(f[x]);
    return f[x];
}
void kruskal()///求次小生成树
{
    sort(p,p+m,cmp);
    for(int i=0; i<=n; i++)
    {
        g[i].clear();
        g[i].push_back(i);
        f[i]=i;
    }
    int sum=0,k=0;///最小生成树权值
    for(int i=0; i<m; i++)
    {
        if(k==n-1)break;
        int x1=fa(p[i].a),x2=fa(p[i].b);
        if(x1!=x2)
        {
            k++;
            p[i].vis=1;
            sum+=p[i].w;
            int l=g[x1].size(),ll=g[x2].size();
            for(int j=0; j<l; j++)
                for(int k=0; k<ll; k++)
                    len[g[x1][j]][g[x2][k]]=len[g[x2][k]][g[x1][j]]=p[i].w;
            f[x1]=x2;
            int tem[110];
            for(int j=0; j<ll; j++)
                tem[j]=g[x2][j];
            for(int j=0; j<l; j++)
                g[x2].push_back(g[x1][j]);
            for(int j=0; j<ll; j++)
                g[x1].push_back(tem[j]);
        }
    }
    int cisum=oo;///次小生成树权值
    for(int i=0; i<m; i++)
        if(!p[i].vis)
            cisum=min(cisum,sum+p[i].w-len[p[i].a][p[i].b]);
}


  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值