计蒜客——灌溉(最小生成树)

时间:2022-12-23 21:31:17

题目链接:https://nanti.jisuanke.com/t/34

到了旱季农业生产的灌溉就成了一个大问题。为了保证灌溉的顺利,某县*决定投资为各个村之间建立灌溉管道。

输入第1行包括一个整数N,表示某县的村庄的数量。(3≤N≤100),第2行-结尾为一个N×N的矩阵,表示每个村庄之间的距离。虽然在理论上,他们是N行,每行由N个用空格分隔的数组成,实际上,他们限制在80个字符,因此,某些行会紧接着另一些行。当然,对角线将会是0,因为不会有线路从第i个村到它本身(任何两个村之间的距离都不大于100000)。

输出只有一个,为修建灌溉管道将所有村庄相连在一个灌溉系统里所需的最小管道长度。

样例输入

4
0 4 9 21
4 0 8 17
9 8 0 16
21 17 16 0
样例输出

28

【思路分析】裸的最小生成树。
【AC代码 】

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

struct node
{
int u,v,cost;
}p[10005];

int pre[105];
int fin(int x)
{
if(x==pre[x])
{
return x;
}
else
{
return pre[x]=fin(pre[x]);
}
}

void join(int x,int y)
{
int t1=fin(x);
int t2=fin(y);
if(t1!=t2)
{
pre[t1]=t2;
}
}

bool cmp(node a,node b)
{
return a.cost<b.cost;
}

int main()
{
int n;
while(~scanf("%d",&n))
{
for(int i=0;i<=n;i++)
{
pre[i]=i;
}
int num,sum=0;
for(int i=0;i<n;i++)
{
for(int j=0;j<n;j++)
{
scanf("%d",&num);
if(i!=j)
{
p[sum].u=i;
p[sum].v=j;
p[sum++].cost=num;
}
}
}
sort(p,p+sum,cmp);
int re=0,ss=0;
for(int i=0;i<sum;i++)
{
if(fin(p[i].u)!=fin(p[i].v))
{
join(p[i].u,p[i].v);
ss++;
re+=p[i].cost;
}
if(ss==n-1)
{
break;
}
}
printf("%d\n",re);
}
return 0;
}