Java语言中的Kruskal算法
Kruskal算法是一种贪婪算法,其工作原理如下:
1.它在图形中创建一组所有边。
2.虽然上述集合不是空的,并且并非所有顶点都被覆盖,
它从该组中删除最小重量边
它检查此边缘是形成一个循环还是仅连接2棵树。如果形成一个循环,则丢弃该边,否则将其添加到树中。
3.完成上述处理后,我们便有了最小生成树。
为了实现此算法,我们需要2个以上的数据结构。
首先,我们需要一个优先级队列,我们可以使用它来使边缘保持排序,并在每次迭代中获得所需的边缘。
接下来,我们需要一个不相交的集合数据结构。不相交集数据结构(也称为联合查找数据结构或合并查找集)是一种数据结构,该数据结构跟踪被划分为多个不相交(不重叠)子集的元素集。每当我们将新节点添加到树中时,我们都会检查它们是否已连接。如果是,那么我们有一个周期。如果否,我们将使边缘的两个顶点并合。这会将它们添加到相同的子集中。
让我们看一下UnionFind或DisjointSet数据结构&minsu;的实现。
示例
class UnionFind { constructor(elements) { //断开连接的组件数量 this.count = elements.length; //跟踪连接的组件 this.parent = {}; //初始化数据结构,使所有 //元素有自己的父母 elements.forEach(e => (this.parent[e] = e)); } union(a, b) { let rootA = this.find(a); let rootB = this.find(b); //根是相同的,因此它们已经连接。 if (rootA === rootB) return; //始终将根较小的元素作为父元素。 if (rootA < rootB) { if (this.parent[b] != b) this.union(this.parent[b], a); this.parent[b] = this.parent[a]; } else { if (this.parent[a] != a) this.union(this.parent[a], b); this.parent[a] = this.parent[b]; } } //返回节点的最终父级 find(a) { while (this.parent[a] !== a) { a = this.parent[a]; } return a; } //检查2个节点的连接 connected(a, b) { return this.find(a) === this.find(b); } }
您可以使用以下方式进行测试:
示例
let uf = new UnionFind(["A", "B", "C", "D", "E"]); uf.union("A", "B"); uf.union("A", "C"); uf.union("C", "D"); console.log(uf.connected("B", "E")); console.log(uf.connected("B", "D"));
输出结果
这将给出输出-
false true
现在让我们看看使用此数据结构的Kruskal算法的实现-
示例
kruskalsMST() { //初始化将包含MST的图形 const MST = new Graph(); this.nodes.forEach(node => MST.addNode(node)); if (this.nodes.length === 0) { return MST; } //创建一个优先级队列 edgeQueue = new PriorityQueue(this.nodes.length * this.nodes.length); //将所有边添加到队列中: for (let node in this.edges) { this.edges[node].forEach(edge => { edgeQueue.enqueue([node, edge.node], edge.weight); }); } let uf = new UnionFind(this.nodes); //循环直到我们探索所有节点或队列为空 while (!edgeQueue.isEmpty()) { //使用解构获取边缘数据 let nextEdge = edgeQueue.dequeue(); let nodes = nextEdge.data; let weight = nextEdge.priority; if (!uf.connected(nodes[0], nodes[1])) { MST.addEdge(nodes[0], nodes[1], weight); uf.union(nodes[0], nodes[1]); } } return MST; }
您可以使用以下方式进行测试:
示例
let g = new Graph(); g.addNode("A"); g.addNode("B"); g.addNode("C"); g.addNode("D"); g.addNode("E"); g.addNode("F"); g.addNode("G"); g.addEdge("A", "C", 100); g.addEdge("A", "B", 3); g.addEdge("A", "D", 4); g.addEdge("C", "D", 3); g.addEdge("D", "E", 8); g.addEdge("E", "F", 10); g.addEdge("B", "G", 9); g.addEdge("E", "G", 50); g.kruskalsMST().display();
输出结果
这将给出输出-
A->B, D B->A, G C->D D->C, A, E E->D, F F->E G->B