Prim算法 数组实现

public class Test {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
//		 带权路径图1
		int 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
		// 带权路径图2
		int 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集合中点的数目达到6
		while(getSetNumber(set) < 6){
//			System.out.println(1);
		// 遍历集合中的边 获得最小权值的边 放入集合 设置result的权值
			int pow = 9999;
			//从第一个元素开始遍历 set[k]为一个顶点的下标+1
			for (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;
				}
			}
		}

	}

}