您好,欢迎访问一九零五行业门户网

如何使用java实现图的最小生成树算法

如何使用java实现图的最小生成树算法
概念介绍:
最小生成树(minimum spanning tree, mst)是指在一个带权有向图或无向图中,找到一棵树,使得其包含图中的所有顶点且权值之和最小。最小生成树算法有多种,其中最经典的两种算法分别是prim算法和kruskal算法。
prim算法:
prim算法是一种基于点的贪婪算法,它从一个顶点开始,然后逐渐扩展,直到生成整棵最小生成树。下面是使用java实现prim算法的示例代码:
import java.util.arrays;public class primalgorithm { // 表示无穷大 private static final int inf = integer.max_value; public static void primmst(int[][] graph) { int vertices = graph.length; // 创建一个数组用来保存最小生成树的顶点 int[] parent = new int[vertices]; // 创建一个数组用来保存每个顶点与最小生成树的最小权值 int[] key = new int[vertices]; // 创建一个数组用来标记顶点是否已经加入最小生成树 boolean[] mstset = new boolean[vertices]; // 初始化key数组和mstset数组的值 arrays.fill(key, inf); arrays.fill(mstset, false); //将第一个顶点加入最小生成树 key[0] = 0; parent[0] = -1; for (int count = 0; count < vertices - 1; count++) { // 选择key值最小的顶点 int minkey = getminkey(key, mstset); mstset[minkey] = true; // 更新与该顶点相邻的顶点的key值 for (int v = 0; v < vertices; v++) { if (graph[minkey][v] != 0 && !mstset[v] && graph[minkey][v] < key[v]) { parent[v] = minkey; key[v] = graph[minkey][v]; } } } // 输出最小生成树 printmst(parent, graph); } // 获得key值最小的顶点 private static int getminkey(int[] key, boolean[] mstset) { int minkey = inf, minindex = -1; for (int v = 0; v < key.length; v++) { if (!mstset[v] && key[v] < minkey) { minkey = key[v]; minindex = v; } } return minindex; } // 输出最小生成树 private static void printmst(int[] parent, int[][] graph) { system.out.println("edge weight"); for (int i = 1; i < graph.length; i++) { system.out.println(parent[i] + " - " + i + " " + graph[i][parent[i]]); } } public static void main(string[] args) { int[][] graph = {{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); }}
kruskal算法:
kruskal算法是一种基于边的贪婪算法,它按照权值从小到大的顺序选择边,并且只选择不会产生环的边,直到生成整棵最小生成树。下面是使用java实现kruskal算法的示例代码:
import java.util.*;class edge implements comparable<edge> { int src, dest, weight; public int compareto(edge compareedge) { return this.weight - compareedge.weight; }}class kruskalalgorithm { public list<edge> kruskalmst(list<edge> edges, int vertices) { list<edge> result = new arraylist<>(); collections.sort(edges); int[] parent = new int[vertices]; for (int i = 0; i < vertices; i++) { parent[i] = i; } int count = 0, i = 0; while (count < vertices - 1) { edge currentedge = edges.get(i); int x = find(parent, currentedge.src); int y = find(parent, currentedge.dest); if (x != y) { result.add(currentedge); union(parent, x, y); count++; } i++; } return result; } private int find(int[] parent, int vertex) { if (parent[vertex] != vertex) { parent[vertex] = find(parent, parent[vertex]); } return parent[vertex]; } private void union(int[] parent, int x, int y) { int xset = find(parent, x); int yset = find(parent, y); parent[xset] = yset; } public static void main(string[] args) { int vertices = 4; list<edge> edges = new arraylist<>(); edges.add(new edge(0, 1, 10)); edges.add(new edge(0, 2, 6)); edges.add(new edge(0, 3, 5)); edges.add(new edge(1, 3, 15)); edges.add(new edge(2, 3, 4)); kruskalalgorithm kruskal = new kruskalalgorithm(); list<edge> result = kruskal.kruskalmst(edges, vertices); system.out.println("edge weight"); for (edge edge : result) { system.out.println(edge.src + " - " + edge.dest + " " + edge.weight); } }}
以上是使用java实现prim算法和kruskal算法的示例代码,它们都是实现图的最小生成树算法的经典方法。通过学习和理解这些代码,可以更好地理解和掌握如何使用java实现图的最小生成树算法。
以上就是如何使用java实现图的最小生成树算法的详细内容。
其它类似信息

推荐信息