每日一题 KY148还是畅通工程

最新推荐文章于 2024-10-09 22:28:58 发布

m0_69230978

最新推荐文章于 2024-10-09 22:28:58 发布

阅读量154

点赞数 5

文章标签：算法

本文链接：https://blog.csdn.net/m0_69230978/article/details/136431000

版权

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

输入描述:

    测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 100 )；随后的N(N-1)2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。     当N为0时，输入结束，该用例不被处理。

输出描述:

    对每个测试用例，在1行里输出最小的公路总长度。

示例1

输入

输出

3
5

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
#define N 10000
int father[N];
int height[N];
//最小生成树
//n个顶点n-1条边 每个结点都能连通且权值之和最小
//建立并查集
//所有边按从小到大遍历每条边
//x y属于同一集合 忽略 不在，该边记录到总和中
struct Edge{
    int x;
    int y;
    int weight;
};
//并查集
//初始化 用一个数组father 编号i
void Init(int n){
    for (int i = 1; i <= n; ++i) {
        father[i]=i;
    }
}
//查
int find(int n){
    //找祖先
    if(n!=father[n]){
        //return find[father[n]];//爸爸的爸爸
        //优化 将结点的父亲改为祖先
        father[n]=find(father[n]);
    }
    return father[n];
}

//并
void Union(int x,int y,int weight,int &total){
    // 一个根到另一个根
    x=find(x);
    y=find(y);
    if(x!=y){
        total+=weight;
    }
    if(height[x]<height[y]){
        father[x]=y;
    }else if (height[x] > height[y]) {//x当父亲
        father[y] = x;
    } else {
        father[y] = x;
        ++height[x];
    }
}
bool compare(Edge lhs,Edge rhs){
    return lhs.weight<rhs.weight;
}
int main() {
    int n;
    while (scanf("%d",&n)!=EOF){//第一行
        if(n==0){
            break;
        }
        Init(n);
        vector<Edge> vec;
        for (int i = 0; i < n*(n-1)/2; ++i) {//n*(n-1)/2行数据
            Edge edge;
            scanf("%d%d%d",&edge.x,&edge.y,&edge.weight);//每读一行
            vec.push_back(edge);//边读完
        }
        sort(vec.begin(),vec.end(), compare);//权值大小排序
        int total=0;
        for (int i = 0; i < n*(n-1)/2; ++i) {
            Union(vec[i].x,vec[i].y,vec[i].weight,total);//合并 如果两个已连 忽略
        }
        printf("%d\n",total);
    }
    return 0;
}