java代码
import java.util.Arrays;
public class PrimAlgorithm {
public static void main(String[] args) {
//测试看看图是否创建ok
char[] data= new char[]{'A','B','C','D','E','F','G'};
int verxs = data.length;
//邻接矩阵的关系使用二维数组表示
int[][] weight = new int[][]{
{10000,5,7,10000,10000,10000,2},
{5,10000,10000,9,10000,10000,3},
{7,10000,10000,10000,8,10000,10000},
{10000,9,10000,10000,10000,4,10000},
{10000,10000,8,10000,10000,5,4},
{10000,10000,10000,4,5,10000,6},
{2,3,10000,10000,4,6,10000},
};
//创建MGraph对象
MGraph graph = new MGraph(verxs);
//创建一个MinTree对象
MinTree minTree = new MinTree();
minTree.createGraph(graph,verxs,data,weight);
//输出
minTree.showGraph(graph);
minTree.prim(graph,0);
}
}
//创建最小生成树 -> 村庄的图
class MinTree{
//创建图的邻接矩阵 和图的节点数组
/**
* 给graph 赋值。graph.data graph.weight
* @param graph 图
* @param verxs 图的节点个数
* @param data 图的节点
* @param weight 图的边
*/
public void createGraph(MGraph graph,int verxs,char[] data,int[][] weight){
for (int i = 0; i < verxs; i++) {
graph.data[i] = data[i];
for(int j=0;j<verxs;j++){
graph.weight[i][j] = weight[i][j];
}
}
}
//显示图的邻接矩阵
// 输出weight二维矩阵
public void showGraph(MGraph graph){
for(int[] link:graph.weight){
System.out.println(Arrays.toString(link));
}
}
//编写prim算法,得到最小生成树
/**
*
* @param graph 图
* @param v 表示从图的第几个顶点开始生成‘A' ->0 'B' ->1
*/
public void prim(MGraph graph,int v){
//1 visited[] 标记节点(顶点)是否被访问过 长度与顶点的个数相同,java中默认为0 ,表示没有访问过
int[] visited = new int[graph.verxs];
//2 把当前节点标记为已访问
visited[v] =1;
//3 定义两个索引,用于记录新产生的边的两个顶点,一个是已经访问过的顶点,一个是这次新访问的顶点,初始化都为-1.
int h1 = -1;
int h2 = -1;
//4 将minWeight 初始化成一个大数,后面在遍历过程中,会被替换,因为要找小的,所以要初始化一个大数。
int minWeight = 10000;
//5 循环遍历 verxs -1次,寻找边,verxs个顶点有verxs-1 条边
for (int k = 1; k < graph.verxs; k++) {
//这个是确定每一次生成的子图,和哪个节点的距离最近
//i节点表示被访问过的节点
//j表示还没有访问过的节点
for (int i = 0; i < graph.verxs; i++) {
for (int j = 0; j < graph.verxs; j++) {
//寻找被访问节点中,与未被访问节点的距离的最小值
if(visited[i] ==1 && visited[j]==0 && graph.weight[i][j] < minWeight){
//如果比已经找到的minWeight小,则替换minWeight(寻找已经访问过的节点和未访问过的节点间权值的最小的边)
minWeight = graph.weight[i][j];
//让h1 = 满足条件的已经访问的节点,h2 = 满足条件的这次访问的节点
h1 = i;
h2 = j;
}
}
}
//找到一条边最小
System.out.println("边《"+graph.data[h1]+","+graph.data[h2]+">"+","+minWeight);
//将当前这个节点标记为已经访问
visited[h2] = 1;
//minWeight 重新设置最大值为10000
minWeight = 10000;
}
}
}
//新建图类,保存结点个数、结点数据的一维char数组,二维int数据的weight数组
class MGraph{
//1 表示图的节点个数
int verxs;
//2 存放结点数据
char[] data;
//3 存放边,就是我们的邻接矩阵
int[][] weight;
//4 构造器,含有图结点的个数
public MGraph(int verxs){
// 初始化图结点个数
this.verxs = verxs;
// 初始化节点数组
this.data = new char[verxs];
// 初始化边数组
this.weight = new int[verxs][verxs];
}
}