最小生成树：Prime算法

Prim算法（加点法）

普里姆算法在找最小生成树时，将顶点分为两类，一类是在查找的过程中已经包含在树中的（假设为 A 类），剩下的是另一类（假设为 B 类）。
对于给定的连通网，起始状态全部顶点都归为 B 类。
在找最小生成树时，选定任意一个顶点作为起始点，并将之从 B 类移至 A 类；
然后找出 B 类中到 A 类中的顶点之间权值最小的顶点，将之从 B 类移至 A 类，如此重复，直到 B 类中没有顶点为止。
所走过的顶点和边就是该连通图的最小生成树。

package com.yue.algorithm.pa01;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.Set;
/**
 * 最小生成树
 * @author WenYueGuang
 * @date 2020年6月8日
 */
public class Prime {
	class edge {
		int startVertex;
		int endVertex;
		int weight;// 权值

		public edge(int startVertex,int endVertex,int weight){
			this.startVertex=startVertex;
			this.endVertex=endVertex;
			this.weight=weight;
		}
	}
	public static void main(String[] args) {
		Prime y = new Prime();
		int n=5;//顶点
		int p=7;//边
		edge[] A=new edge[p];
		A[0]=y.new edge(0, 1, 5);
		A[1]=y.new edge(1, 2, 5);
		A[2]=y.new edge(1, 3, 12);
		A[3]=y.new edge(1, 4, 15);
		A[4]=y.new edge(2, 3, 17);
		A[5]=y.new edge(2, 4, 6);
		A[6]=y.new edge(3, 4, 12);
		y.getMinSpanTree(n, A);
	}
	public ArrayList<edge> getMinSpanTree(int n, edge[] A) {
		ArrayList<edge> rs = new ArrayList<Prime.edge>();
		Set<Integer> unUsed = new HashSet<Integer>();
		Set<Integer> used = new HashSet<Integer>();
		boolean[] usedEdge = new boolean[A.length];
		for(int i=1;i<n;i++) {
			unUsed.add(i);
		}
		used.add(0);		
		int startVertex;
		int endVertex;
		int index = 0;
		while(used.size()<n) {
			edge tmp = null;
			for(int i=0;i<A.length;i++) {
				if(usedEdge[i])continue;
				startVertex = A[i].startVertex;
				endVertex = A[i].endVertex;
				if(((unUsed.contains(startVertex) && used.contains(endVertex))
						|| (unUsed.contains(endVertex) && used.contains(startVertex)))
						&& (tmp==null || tmp.weight>A[i].weight)) {
					tmp=A[i];
					index = i;
				}
			}
			if(tmp==null) {
				break;
			}
			startVertex = tmp.startVertex;
			endVertex = tmp.endVertex;
			if(unUsed.contains(startVertex)){
				used.add(startVertex);
				unUsed.remove(startVertex);
			}else {
				used.add(endVertex);
				unUsed.remove(endVertex);
			}
			rs.add(tmp);
			usedEdge[index] = true;
		}
		
		for (edge ed : rs) {
			System.out.println(ed.startVertex+"---"+ed.endVertex+"---"+ed.weight);
		}
		return rs;
	}	
}