题解 | #Jungle Roads#

Jungle Roads

https://www.nowcoder.com/practice/75c19b92d6b942f08989b335afbc73c3

#include <bits/stdc++.h>

using namespace std;

struct Edge {
    int a;
    int b;
    int weigh;

    Edge(char _a, char _b, int _weigh) {
        a = _a - 'A';
        b = _b - 'A';
        weigh = _weigh;
    }
};


bool operator<(Edge lhs, Edge rhs) {
    return lhs.weigh < rhs.weigh;
}

int father[100];
int height[100];

void InitSet(int n) {
    for (int i = 0; i < n; ++i) {
        father[i] = i;
        height[i] = 0;
    }
}

int FindFather(int u) {
    if (u == father[u]) {
        return father[u];
    } else {
        father[u] = FindFather(father[u]);
        return father[u];
    }
}

void UnionSet(int u, int v) {
    int u_root = FindFather(u);
    int v_root = FindFather(v);
    if (height[u_root] > height[v_root]) {
        father[v_root] = u_root;
    } else if (height[v_root] > height[u_root]) {
        father[u_root] = v_root;
    } else {
        father[v_root] = u_root;
        ++height[u_root];
    }
}

int Kruskal(int n, vector<Edge> &edgeVec) {
    InitSet(n);
    int sum = 0;
    int edgeNum = 0;
    sort(edgeVec.begin(), edgeVec.end());
    for (int i = 0; i < edgeVec.size(); ++i) {
        Edge a = edgeVec[i];
        if (FindFather(a.a) != FindFather(a.b)) {
            UnionSet(a.a, a.b);
            edgeNum++;
            sum += a.weigh;
            if (edgeNum == n - 1) {
                break;
            }
        }
    }
    return sum;
}

int main() {
    int n;//村庄的个数
    while (scanf("%d", &n) != EOF) {
        if (0 == n) {
            return 0;
        }
        vector<Edge> edgeVec;
        for (int i = 0; i < n - 1; ++i) {
            char a;
            cin >> a;
            int m;
            scanf("%d", &m);
            for (int j = 0; j < m; ++j) {
                char b;
                cin >> b;
                int weigh;
                scanf("%d", &weigh);
                Edge e(a, b, weigh);
                edgeVec.push_back(e);
            }
        }
        printf("%d\n", Kruskal(n, edgeVec));

    }
    return 0;
}

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务