Geeks : Kruskal’s Minimum Spanning Tree Algorithm 最小生成树
版权声明:本文作者靖心,靖空间地址:http://blog.csdn.net/kenden23/。未经本作者同意不得转载。
https://blog.csdn.net/kenden23/article/details/26821635
寻找图中最小连通的路径,图例如以下:
算法步骤:
1. Sort all the edges in non-decreasing order of their weight. 2. Pick the smallest edge. Check if it forms a cycle with the spanning tree formed so far. If cycle is not formed, include this edge. Else, discard it. 3. Repeat step#2 until there are (V-1) edges in the spanning tree.关键是第二步难,这里使用Union Find来解决,能够差点儿小于O(lgn)的时间效率来推断是否须要推断的顶点和已经选择的顶点成环。
正由于这步,使得原本简单的贪心法。变得不那么简单了。
这样本算法的时间效率达到:max(O(ElogE) , O(ElogV))
原文參考:http://www.geeksforgeeks.org/greedy-algorithms-set-2-kruskals-minimum-spanning-tree-mst/
#pragma once
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
class KruskalsMST
{
struct Edge
{
int src, des, weight;
};
static int cmp(const void *a, const void *b)
{
Edge *a1 = (Edge *) a, *b1 = (Edge *) b;
return a1->weight - b1->weight;
}
struct Graph
{
int V, E;
Edge *edges;
Graph(int v, int e) : V(v), E(e)
{
edges = new Edge[e];
}
virtual ~Graph()
{
if (edges) delete [] edges;
}
};
struct SubSet
{
int parent, rank;
};
int find(SubSet *subs, int i)
{
if (subs[i].parent != i)
subs[i].parent = find(subs, subs[i].parent);
return subs[i].parent;
}
void UnionTwo(SubSet *subs, int x, int y)
{
int xroot = find(subs, x);
int yroot = find(subs, y);
if (subs[xroot].rank < subs[yroot].rank)
subs[xroot].parent = yroot;
else if (subs[xroot].rank > subs[yroot].rank)
subs[yroot].parent = xroot;
else
{
subs[xroot].rank++;
subs[yroot].parent = xroot;
}
}
Graph *graph;
Edge *res;
SubSet *subs;
void initSubSet()
{
subs = new SubSet[graph->V];
for (int i = 0; i < graph->V; i++)
{
subs[i].parent = i;
subs[i].rank = 0;
}
}
void mst()
{
res = new Edge[graph->V-1];
qsort(graph->edges, graph->E, sizeof(graph->edges[0]), cmp);
initSubSet();
for (int e = 0, i = 0; e < graph->V - 1 && i < graph->E; i++)
{
Edge nextEdge = graph->edges[i];
int x = find(subs, nextEdge.src);
int y = find(subs, nextEdge.des);
if (x != y)
{
res[e++] = nextEdge;
UnionTwo(subs, x, y);
}
}
}
void printResult()
{
printf("Following are the edges in the constructed MST\n");
for (int i = 0; i < graph->V-1; ++i)
printf("%d -- %d == %d\n", res[i].src, res[i].des, res[i].weight);
}
public:
KruskalsMST()
{
/* Let us create following weighted graph
10
0--------1
| \ |
6| 5\ |15
| \ |
2--------3
4 */
int V = 4; // Number of vertices in graph
int E = 5; // Number of edges in graph
graph = new Graph(V, E);
// add edge 0-1
graph->edges[0].src = 0;
graph->edges[0].des = 1;
graph->edges[0].weight = 10;
// add edges 0-2
graph->edges[1].src = 0;
graph->edges[1].des = 2;
graph->edges[1].weight = 6;
// add edges 0-3
graph->edges[2].src = 0;
graph->edges[2].des = 3;
graph->edges[2].weight = 5;
// add edges 1-3
graph->edges[3].src = 1;
graph->edges[3].des = 3;
graph->edges[3].weight = 15;
// add edges 2-3
graph->edges[4].src = 2;
graph->edges[4].des = 3;
graph->edges[4].weight = 4;
mst();
printResult();
}
~KruskalsMST()
{
if (res) delete [] res;
if (subs) delete [] subs;
if (graph) delete graph;
}
};