并查集的应用 ---最小生成树-Kruskal算法 + 1929 - 最短网络 ---最小生成树

最新推荐文章于 2024-07-24 20:44:05 发布

chlhoa

最新推荐文章于 2024-07-24 20:44:05 发布

阅读量82

点赞数

文章标签：算法 c++ 图论

本文链接：https://blog.csdn.net/weixin_73739312/article/details/129035113

版权

// 并查集的应用 ---最小生成树
/*  Kruskal算法 */
基本思想：
按照权值从小到大的顺序选择n-1条边，并保证这n-1条边不构成回路；
直至连成一颗树为止。

//  1929 - 最短网络 Agri-Net（USACO3.1）  ---最小生成树
//  来源：东方博宜oj  oj.czos.cn
#include<bits/stdc++.h>
using namespace std;
int n;
int f[110];
struct node
{
    int x,y,len;
}a[6000];
int findd(int x)
{
    return x==f[x]?x:f[x]=findd(f[x]);
}
bool cmp(node n1,node n2)
{
    return n1.len<n2.len;
}
int main()
{
    int t,k=0;  // k记录下标
    scanf("%d",&n);
    for(int i=1;i<=n;i++) f[i]=i; //父节点数组f[]初始化
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            scanf("%d",&t);
            if(i<j)         //只要存储对角线右上部分或左下部分的数据即可
            {
                a[++k].x=i;
                a[k].y=j;
                a[k].len=t;
            }
        }
    }
    sort(a+1,a+1+k,cmp);
    int s=0;  // 最短路径的和
    int c=0;  // 修了几条路
    for(int i=1;i<=k;i++)
    {
        // 如果两个点之间本身没有路，就可以修
        // 否则不能修，会构成回路
        int fx=findd(a[i].x);
        int fy=findd(a[i].y);
        if(fx!=fy)
        {
            f[fx]=fy;
            s+=a[i].len;
            c++;
        }
        if(c==n-1)
        {
            cout<< s;
            break;
        }
    }
    return 0;
}