Kruskal(布鲁斯克尔）算法 模板 洛谷P1546 Agri-Net【例4-11】最短网络(agrinet)

【例4-11】最短网络(agrinet)

时间限制: 1000 ms         内存限制: 65536 KB
提交数: 309     通过数: 205 

【题目描述】

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

【输入】

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

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

【输出】

只有一个输出，其中包含连接到每个农场的光纤的最小长度。

【输入样例】

4
0  4  9  21
4  0  8  17
9  8  0  16
21 17 16  0

【输出样例】

28

【代码实现】

#include<bits/stdc++.h>
using namespace std;
const int N=10000;
vector<int>g[N];
int fa[N];
struct node
{
    int x,y,v;
}a[N];

int find(int x)
{
    return x==fa[x]?x:fa[x]=find(fa[x]);
}
int cmp(node x,node y)
{
    return x.v<y.v;
}
int main()
{
   int n,x,m=0;
   scanf("%d",&n);
   for(int i=1;i<=n;i++)
    for(int j=1;j<=n;j++)//转化
   {
       scanf("%d",&x);
       if(x!=0)
       {
           m++;
           a[m].x=i;
           a[m].y=j;
           a[m].v=x;
       }
   }
   for(int i=0;i<=n;i++)
    fa[i]=i;
   sort(a+1,a+m+1,cmp);
   int tot=0,k=0;
   for(int i=1;i<=m;i++)
   {
       int r1=find(a[i].x);
       int r2=find(a[i].y);
       if(r1!=r2)  //不再一个集合
       {
           fa[r1]=r2;
           tot+=a[i].v;
           k++;
       }
       if(k==n-1) break;//找到n-1条边
   }
   printf("%d\n",tot);
  return 0;
}