(Java数据结构和算法)最小生成树---prime算法

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ccnuacmhdu/article/details/84453773

参考博文


public class Main {

	public static void main(String[] args){
		int inf = 1000000;//无穷大
		//图,可以这样认为:图的任意两个顶点之间都有边,两顶点无法到达的,可以认为他们之间的边权是无穷大
		int[][] graph = {
				{inf, 6, 1, inf},
				{6, inf, 5, 3},	
				{1, 5, inf, 4},
				{inf, 3, 4, inf}
			};
		boolean[] visited = new boolean[4];;
		for(int i = 0; i < 4; i++){
			visited[i] = false;
		}
		
		int index = 0;
		int sum = 0;
		int[] distance = new int[4];
		visited[0] = true;//从1这个顶点开始
		for(int i = 0; i < 4; i++){
			distance[i] = graph[0][i];
		}
		//由于已经访问过一个节点了,剩下3个节点,每次能找好一个节点,最多再找3次
		for(int i = 0; i < 3; i++){
			int minimum = inf;
			//从所有没有访问的节点中找一个到已经访问的节点距离最小的
			for(int j = 0; j < 4; j++){
				if(visited[j] == false && distance[j] < minimum){
					minimum = distance[j];
					index = j;
				}
			}
			visited[index] = true;
			sum += minimum;
			//更新距离,上面又新产生了一个访问过的节点,现在更新所有未访问节点
			//到所有已经访问节点的最小距离
			for(int j = 0; j < 4; j++){
				if(visited[j] == false && distance[j] > graph[index][j]){
					distance[j] = graph[index][j];
				}
			}
		}
		System.out.println(sum);
	}
}

猜你喜欢

转载自blog.csdn.net/ccnuacmhdu/article/details/84453773