NC159 最小生成树
连通图中的每一棵生成树,都是原图的一个极大无环子图。也就是说,从其中删去任何一条边,生成树就不再连通;反之,在其中引入任意一条新边,都会形成(恰好)一个回路。
对于一个带权图(即网络)而言,不同的生成树所对应的总权值也不尽相同。那么如何找出总权值最小的生成树呢?
构造最小生成树的方法有多种,典型的有两种:Kruskal算法和Prim算法。
Kruskal算法和Prim算法的原理:最小生成树(Kruskal(克鲁斯卡尔)和Prim(普里姆))算法动画演示
/*
* Kruskal 最小生成树算法
*/
#include<vector>
#include<iostream>
#include<algorithm>
using namespace std;
struct Edge {
int u, v, w;
bool operator<(const Edge& a)const { return w < a.w; }
};
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
* 返回最小的花费代价使得这n户人家连接起来
* @param n int n户人家的村庄
* @param m int m条路
* @param cost intvector<vector<>> 一维3个参数,表示连接1个村庄到另外1个村庄的花费的代价
* @return int
*/
int miniSpanningTree(int n, int m, vector<vector<int>>& cost) {
// write code here
vector<int> parent(n + 1, -1);
vector<Edge> edges;
for (auto tmp : cost) {
Edge e;
e.u = tmp[0];
e.v = tmp[1];
e.w = tmp[2];
edges.push_back(e);
}
int res = kruskal(edges, parent);
}
int findRoot(vector<int>& parent, int x) {
int root = x;
while (parent[root] != -1) {
root = parent[root];
}
return root;
}
void unionNode(vector<int>& parent, int x, int y) {
int x_root = findRoot(parent, x);
int y_root = findRoot(parent, y);
parent[x_root] = y_root;
}
int kruskal(vector<Edge>& edges, vector<int>& parent) {
sort(edges.begin(), edges.end());
int ans = 0;
while (!edges.empty()) {
if (findRoot(parent, edges[0].u) != findRoot(parent, edges[0].v)) {
ans += edges[0].w;
unionNode(parent, edges[0].u, edges[0].v);
}
edges.erase(edges.begin());
}
return ans;
}
};
参考资料
[1]《数据结构(用面向对象方法与C++语言描述)》殷人昆
[2]最小生成树(Kruskal(克鲁斯卡尔)和Prim(普里姆))算法动画演示