JAVA中Prime算法的原理和实现
Prime算法:是一种穷举查找算法来从一个连通图中构造一棵最小生成树。利用始终找到与当前树中节点权重最小的边,找到节点,加到最小生成树的节点集合中,直至所有节点都包括其中,这样就构成了一棵最小生成树。prime在算法中属于贪心算法的一种,贪心算法还有:Kruskal、Dijkstra以及哈夫曼树及编码算法。
下面具体讲一下prime算法:
1、首先需要构造一颗最小生成树,以及两个节点之间的权重数组,在此我们用一个二维数组来代表这样一个连通图的形式。节点就是0~数组长度-1,10000代表节点本身,权重 >= 100代表两个节点不连通,反之连通。
构建连通图代码如下:
// 初始化连通图
public static void initGraph(int[][] graph, ArrayList<Integer> points) {
for(int i = 0 ; i < graph.length; i++) {
points.add(i);
for(int j = 0; j < graph[i].length; j++) {
if(i == j) {
graph[i][j] = 10000;
}else {
int temp = (int)(Math.random() * 200 +1);
graph[i][j] = temp; // 大于等于100不连通, 小于100连通
}
graph[j][i] = graph[i][j];
}
}
}
连通图的数组表示:
2、找到距离当前树中节点权重最小的边,开始节点随机产生,(算法的重点)!!!
// prime算法实现
public static int prime(int[][] graph, ArrayList<Integer> points, int current) {
String path = "";
ArrayList<Integer> selectPoints = new ArrayList<Integer>(); // 选中的点集合
int totalWeights = 0; // 权重总和
selectPoints.add(current); // 添加初始开始节点
points.remove(current); // 从未选择的节点集合中删除被选中的节点
path = "|" + current + "|";
System.out.println("当前路径:" + path);
System.out.println("当前已选中节点: " + selectPoints.toString());
System.out.println("当前剩余节点: " + points.toString());
System.out.println("当前总权重: " &
iApp123: 我也万州的 可否留个联系方式 一起交流学习
iApp123: 老哥 万州的呀
Mr.洛洛: 喜换请点赞哇。