Kruskal 算法 和 Prim算法 是 最小生成树基础的两种算法:
本文章中的代码用c语言写的(不过头文件用了c++ 的,大家不要在意), 其实在 Kruskal 算法 里面对数据进行排序的时候 可以用 C++ 库函数 的 sort排序进行排序 ,不过这里面我写了一个 快排,用快排进行排序了! 在 Kruskal 算法 里面,用到了并查集的 查询 和 合并 两个函数,主要是用来检查某个节点是否用过!
Prim算法 如果你仔细研究的话你会发现,Prim算法 和 最短路 的 Dijkstra 的算法的思想比较相似,基本可以说Prim 运用了 最短路 Dijkstra 的思想!
先给出一组测试数据:
输入第一行有两个数 n 和 m,n 表示有n个城市,m表示有m条道路。接下来的m行,每行形容“a b c”用来表示一条路,意思是城市a 到城市 b 的路径长度为 c。
输入:
6 9
2 4 11
3 5 13
4 6 3
5 6 4
2 3 6
4 5 7
1 2 1
3 4 9
1 3 2
输出结果:
19
// Kruskal 算法: 这里的 Kruskal 算法 的时间复杂度是 O(nlog(n))
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <cstdio>
using namespace std;
struct edge
{
int u;
int v;
int w;
};
struct edge e[110];
int n,m;
int f[7] = {0},sum = 0,Count = 0;
void quicksort(int left,int right) // 快速排序
{
int i,j;
struct edge t;
if(left > right)
return;
i = left;
j = right;
while(i != j)
{
while(e[j].w >= e[left].w && i < j)
j--;
while(e[i].w <= e[left].w && i < j)
i++;
if(i < j)
{
t = e[i];
e[i] = e[j];
e[j] = t;
}
}
t = e[left];
e[left] = e[i];
e[i] = t;
quicksort(left,i - 1);
quicksort(i + 1,right);
return;
}
int getf(int v) // 并查集查找根节点
{
if(f[v] == v)
return v;
else
{
f[v] = getf(f[v]);
return f[v];
}
}
int merge(int v,int u) // 将两棵树合并
{
int t1,t2;
t1 = getf(v);
t2 = getf(u);
if(t1 != t2)
{
f[t2] = t1;
return 1;
}
return 0;
}
int main()
{
int i;
scanf("%d%d",&n,&m);
for(i = 1;i <= m;i++)
{
scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
}
quicksort(1,m); // 按照权值从小到大对边进行快速排序
for(i = 1;i <= n;i++) // 并查集根节点的初始化
{
f[i] = i;
}
for(i = 1;i <= m;i++) // 权值已经按照从小到大排序好了,所以可以直接将不在同一颗树的两颗树合并成一颗树
{
if(merge(e[i].u,e[i].v))
{
Count++;
sum += e[i].w;
}
if(Count == n - 1) // 直到合并到 n - 1 次 ,也是就将 n 棵树树合并成一棵树,跳出循环,这个树就是最小生成树
break;
}
printf("%d\n",sum);
return 0;
}
// Prim算法 该代码 的时间复杂度 为 O(n^2) 如果借助“堆”,然后使用邻接表来储存图的话时间可达到 O(nlog(n))
#include <iostream>
#include <algorithm>
#include <cstring>
#include <string>
#include <cstdio>
using namespace std;
int main()
{
int n,m,i,j,k,min,t1,t2,t3;
int e[7][7],dis[7],book[7] = {0};
int inf = 99999999;
int count = 0,sum = 0;
scanf("%d%d",&n,&m);
for(i = 0;i <= n;i++) // 进行一个初始化
{
for(j = 1;j <= n;j++)
{
if(i == j) // 自己到自己 权为 0
e[i][j] = 0;
else
e[i][j] = inf; // 自己到其他 权为 inf
}
}
for(i = 1;i <= m;i++) // 根据题里给的条件 更新相应的权值
{
scanf("%d%d%d",&t1,&t2,&t3);
e[t1][t2] = t3;
e[t2][t1] = t3;
}
// 初始化dis数组,这里是1号顶点到各个顶点的初始距离,因为当前生成树中只有1号顶点
for(i = 1;i <= n;i++)
{
dis[i] = e[1][i];
}
book[1] = 1; // book 标记该顶点是否已经加入生成树
count++;
while(count < n)
{
min = inf;
for(i = 1;i <= n;i++) // 寻找生成树到 i 节点的最小长度的路径
{
if(book[i] == 0 && dis[i] < min)
{
min = dis[i];
j = i;
}
}
book[j] = 1; // 找到后将 i 节点接到 生成树上
count++;
sum += dis[j]; // 总路径长度加上新增加的长度,更新总长度
// 扫描当前顶点 j 所有的边,再以j为中间点,更新生成树到每一个非树顶点的距离
for(k = 1;k <= n;k++)
{
if(book[k] == 0 && dis[k] > e[j][k])
dis[k] = e[j][k];
}
}
printf("%d\n",sum);
return 0;
}