用C语言编程实现构建最小生成树的两种算法,可以动态输入,并详细解释代码
时间: 2023-12-21 09:03:57 浏览: 68
好的,下面我将介绍Prim算法和Kruskal算法,它们都是构建最小生成树的常用算法,同时也是经典的图论算法之一。
### Prim算法
Prim算法是一种贪心算法,它的基本思想是从一个点开始,每次选择与已有的生成树中距离最近的点加入到生成树中。
下面是Prim算法的C语言代码实现:
```c
#include <stdio.h>
#include <limits.h>
#define V 5
// 找到最小权值的顶点
int minKey(int key[], bool mstSet[])
{
int min = INT_MAX, min_index;
for (int v = 0; v < V; v++)
if (mstSet[v] == false && key[v] < min)
min = key[v], min_index = v;
return min_index;
}
// 打印生成的最小生成树
void printMST(int parent[], int graph[V][V])
{
printf("Edge \tWeight\n");
for (int i = 1; i < V; i++)
printf("%d - %d \t%d \n", parent[i], i, graph[i][parent[i]]);
}
void primMST(int graph[V][V])
{
int parent[V]; // 保存最小生成树
int key[V]; // 保存每个顶点的权值
bool mstSet[V]; // 记录每个顶点是否已经被访问过
// 初始化所有的key值为最大值
for (int i = 0; i < V; i++)
key[i] = INT_MAX, mstSet[i] = false;
key[0] = 0;
parent[0] = -1;
for (int count = 0; count < V - 1; count++) {
int u = minKey(key, mstSet);
mstSet[u] = true;
for (int v = 0; v < V; v++)
if (graph[u][v] && mstSet[v] == false && graph[u][v] < key[v])
parent[v] = u, key[v] = graph[u][v];
}
printMST(parent, graph);
}
int main()
{
int graph[V][V] = {
{ 0, 2, 0, 6, 0 },
{ 2, 0, 3, 8, 5 },
{ 0, 3, 0, 0, 7 },
{ 6, 8, 0, 0, 9 },
{ 0, 5, 7, 9, 0 },
};
primMST(graph);
return 0;
}
```
上述代码中,我们通过一个二维数组`graph`来表示一个图,其中`V`是顶点数。
在`primMST`函数中,我们首先初始化了所有的顶点的权值为最大值,然后将第一个点作为起点。
接下来,我们通过`minKey`函数找到与已有生成树中距离最近的顶点。然后将这个顶点加入到已有的生成树中,并更新所有未被访问过的顶点的权值。
最后,我们通过`printMST`函数打印出最小生成树的边和权值。
### Kruskal算法
Kruskal算法也是一种贪心算法,它的基本思想是从边入手,每次选择一条权值最小且不会形成环的边加入到生成树中。
下面是Kruskal算法的C语言代码实现:
```c
#include <stdio.h>
#include <stdlib.h>
#define V 5
#define E 7
struct Edge {
int src, dest, weight;
};
struct Graph {
int V, E;
struct Edge* edges;
};
struct Graph* createGraph(int V, int E)
{
struct Graph* graph = (struct Graph*) malloc(sizeof(struct Graph));
graph->V = V;
graph->E = E;
graph->edges = (struct Edge*) malloc(graph->E * sizeof(struct Edge));
return graph;
}
// 找到顶点所在的集合
int find(int parent[], int i)
{
if (parent[i] == -1)
return i;
return find(parent, parent[i]);
}
// 合并两个集合
void Union(int parent[], int x, int y)
{
int xset = find(parent, x);
int yset = find(parent, y);
parent[xset] = yset;
}
// 比较边的权值
int myComp(const void* a, const void* b)
{
struct Edge* a1 = (struct Edge*) a;
struct Edge* b1 = (struct Edge*) b;
return a1->weight > b1->weight;
}
void kruskalMST(struct Graph* graph)
{
int* parent = (int*) malloc(graph->V * sizeof(int));
for (int i = 0; i < graph->V; i++)
parent[i] = -1;
qsort(graph->edges, graph->E, sizeof(graph->edges[0]), myComp);
struct Edge result[graph->V];
int e = 0, i = 0;
while (e < graph->V - 1 && i < graph->E) {
struct Edge next_edge = graph->edges[i++];
int x = find(parent, next_edge.src);
int y = find(parent, next_edge.dest);
if (x != y) {
result[e++] = next_edge;
Union(parent, x, y);
}
}
printf("Edge \tWeight\n");
for (i = 0; i < e; ++i)
printf("%d - %d \t%d \n", result[i].src, result[i].dest, result[i].weight);
return;
}
int main()
{
struct Graph* graph = createGraph(V, E);
graph->edges[0].src = 0;
graph->edges[0].dest = 1;
graph->edges[0].weight = 2;
graph->edges[1].src = 0;
graph->edges[1].dest = 3;
graph->edges[1].weight = 6;
graph->edges[2].src = 1;
graph->edges[2].dest = 3;
graph->edges[2].weight = 8;
graph->edges[3].src = 1;
graph->edges[3].dest = 2;
graph->edges[3].weight = 3;
graph->edges[4].src = 1;
graph->edges[4].dest = 4;
graph->edges[4].weight = 5;
graph->edges[5].src = 2;
graph->edges[5].dest = 4;
graph->edges[5].weight = 7;
graph->edges[6].src = 3;
graph->edges[6].dest = 4;
graph->edges[6].weight = 9;
kruskalMST(graph);
return 0;
}
```
上述代码中,我们首先定义了一个`struct Edge`结构体来表示图中的边,以及一个`struct Graph`结构体来表示一个图。
在`kruskalMST`函数中,我们首先将所有的边按照权值进行排序,然后逐个选择边加入到生成树中,直到生成树的边数等于`V-1`为止。
在选择边的过程中,我们通过`find`函数找到每个顶点所在的集合,如果两个顶点不在同一个集合中,则将它们合并到一个集合中,并将这条边加入到生成树中。
最后,我们通过`result`数组打印出生成树的边和权值。
阅读全文