CODEVS1078最小生成树

Posted 2020-06-13

Description

农民约翰被选为他们镇的镇长！他其中一个竞选承诺就是在镇上建立起互联网，并连接到所有的农场。当然，他需要你的帮助。 约翰已经给他的农场安排了一条高速的网络线路，他想把这条线路共享给其他农场。为了使花费最少，他想铺设最短的光纤去连接所有的农场。 你将得到一份各农场之间连接费用的列表，你必须找出能连接所有农场并所用光纤最短的方案。 每两个农场间的距离不会超过100000

Input

第一行： 农场的个数，N（3<=N<=100）。

第二行..结尾: 接下来的行包含了一个N*N的矩阵,表示每个农场之间的距离。理论上，他们是N行，每行由N个用空格分隔的数组成，实际上，他们每行限制在80个字符以内，因此，某些行会紧接着另一些行。当然，对角线将会是0，因为线路从第i个农场到它本身的距离在本题中没有意义。

Output

只有一个输出，是连接到每个农场的光纤的最小长度和。

Sample Input

4

0  4  9 21

4  0  8 17

9  8  0 16

21 17 16  0

Sample Output

28

 

#include<iostream>
#include<algorithm>
using namespace std;
const int MAXN=110;
int f[MAXN];
int n,m,v,cnt=0,tot=0,ans=0;
struct point
{
    int x,y,v;
}a[MAXN*MAXN];
int root(int x)
{
    if (x!=f[x]) f[x]=root(f[x]);
    return f[x];
}
void unionn(int a,int b)
{
    f[a]=b;
}
int px(const point &a,const point &b)
{
    return (a.v<b.v);
}
int main()
{
    cin>>n;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        {
            cin>>v;
            if (v)
            {
                cnt++;
                a[cnt].x=i; a[cnt].y=j;a[cnt].v=v;
            }
        }
    sort(a+1,a+cnt+1,px);
    for (int i=1;i<=n;i++) f[i]=i;//初始化 注意不是cnt啊666
    for (int i=1;i<=cnt;i++)
    {
        if (root(a[i].x)!=root(a[i].y))
        {
            unionn(f[a[i].x],f[a[i].y]);
            ++tot;
            ans+=a[i].v;
        }
        if (tot==n-1) break;
    }
    cout<<ans;
    return 0;
}