Prim算法 数组实现

public class Test {public static void main(String[] args) {// TODO Auto-generated method stub// 带权路径图1int net1[][] = {{ 0, 1, 2, 3, 4, 5 },{ 1, 0, 3, 2, 0, 0 },{ 2, 3, 0, 0, 0, 6 },{ 3, 2, 0, 0, 7, 0 },{ 4, 0, 0, 7, 0, 5 }, { 5, 0, 6, 0, 5, 0 } };// 带权路径图2// 带权路径图2int net[][] = {{ 0, 6, 0, 0, 7, 4 },{ 6, 0, 3, 4, 0, 0 },{ 0, 3, 0, 0, 0, 2 },{ 0, 4, 0, 0, 5, 0 },{ 7, 0, 0, 5, 0, 1 }, { 4, 0, 2, 0, 1, 0 } };net = prim(net1);for (int i = 0; i < net.length; i++) {for (int j = 0; j < net.length; j++) {System.out.print(net[i][j]+" ");}System.out.println();}}public static int[][] prim(int net[][]) {int result[][] = new int[net.length][net.length];int x = 0, y = 1;int set[] = new int[net.length];// 初始化结果数组for (int i = 0; i < net.length; i++) {for (int j = 0; j < net.length; j++) {result[i][j] = 0;}}// 将最小权值的两点放入集合for (int i = 0; i < net.length; i++) {for (int j = i + 1; j < net.length; j++) {if (net[i][j] <= net[x][y] && net[i][j] != 0) {x = i;y = j;}}}InsertSort(set, x);InsertSort(set, y);// 更新结果数组的权值(无向）result[x][y] = net[x][y];result[y][x] = net[y][x];// 循环 直到set集合中点的数目达到6while(getSetNumber(set) < 6){//System.out.println(1);// 遍历集合中的边 获得最小权值的边 放入集合 设置result的权值int pow = 9999;//从第一个元素开始遍历 set[k]为一个顶点的下标+1for (int k = 0; k < getSetNumber(set); k++) {//遍历元素的边 i要还原下标 所以-1 因为在set数组中是用下标+1的方式储存 for (int i = set[k]-1; i < set.length; i++) {for (int j = 0; j < set.length; j++) {//只有不与已在集合内的元素的边（+1） 和 权值更小的边 但不为零 可以作为目标if( !isCover(set,j+1) && net[i][j] <= pow && net[i][j] != 0){pow = net[i][j];//System.out.println(pow+" "+i+" "+j);x = i;y = j;}}}}//跟新集合内的元素和结果的权值InsertSort(set, x);InsertSort(set, y);result[x][y] = net[x][y];result[y][x] = net[y][x];//for (int i = 0; i < set.length; i++) {//System.out.print(set[i]+" ");//}//System.out.println();}return result;}//返回数组里非0的数字的个数public static int getSetNumber(int[] set){for (int i = 0; i < set.length; i++) {if(set[i] == 0)return i;if(i + 1 == set.length)return set.length;}return 0;}//判断数字n 是否在set里public static boolean isCover(int[] set, int n) {for (int i = 0; i < set.length; i++) {if (set[i] == n)return true;}return false;}//动态增序插入一个数 如果这个数已经存在在数组中 就不插入public static void InsertSort(int[] set, int num) {num += 1;// 数组中没有数 直接放到第一位if (set[0] == 0)set[0] = num;else {// 如果已经有相等的数 不进行插入for (int i = 0; i < set.length; i++) {if (num == set[i])return;}for (int i = 0; i < set.length; i++) {// 如果插入的数小于第i个数 往后偏移一位 放在第i位if (num < set[i]) {for (int j = set.length - 1; j > i; j--) {set[j] = set[j - 1];}set[i] = num;return;}}// 如果插入的数大于所有数 放在最后for (int i = 0; i < set.length; i++) {if (set[i] == 0) {set[i] = num;return;}}}}}