灌溉（图论+最小生成树+并查集）

最新推荐文章于 2020-05-08 20:53:41 发布

littlefox993

最新推荐文章于 2020-05-08 20:53:41 发布

阅读量262

点赞数

本文链接：https://blog.csdn.net/weixin_42671353/article/details/82558004

版权

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

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

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

样例输入复制

样例输出复制

题解：也是裸的最小生成树。题目不同在于直接给出了图，动点脑筋想想怎么存数据。

代码：

#include<iostream>
#include<algorithm>
using namespace std;
const int maxn = 105;
int Tree[maxn];
struct city
{
    int u;
    int v;
    int cost;
} c[10005];  //注意100*100
bool cmp(city a,city b)
{
    return a.cost<b.cost;
}
int findRoot(int x)
{
    if(Tree[x]==-1)
        return x;
    else
    {
        int tmp=findRoot(Tree[x]);
        Tree[x]=tmp;
        return tmp;
    }
}
int main()
{
    int n;
    cin>>n;
    int tmp;
    for(int i=0; i<n; i++)
        Tree[i]=-1;
    int sum=0;
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<n; j++)
        {
            cin>>tmp;
            if(i!=j)
            {
                c[sum].u=i;
                c[sum].v=j;
                c[sum].cost=tmp;
                sum++;
            }

        }
    }
    sort(c,c+sum,cmp);
    int ans=0;
    for(int i=0; i<sum; i++)
    {
        int aa=findRoot(c[i].u);
        int bb=findRoot(c[i].v);
        if(aa!=bb)
        {
            Tree[aa]=bb;
            ans+=c[i].cost;
        }
    }
    printf("%d\n",ans);
    return 0;
}