Prim算法

解决这个问题我用的是Set集合,因为Set有不加重复元素,又可以迭代其中的元素,以及知道其大小等优点。

我们把树上的所有顶点分成2部分,已经访问的顶点称之为U,还没访问的称之为  V-U,找最小生成树的过程就是 遍历V-U中的点,使他们到U中的某个点距离最小!!

然后再把这个点加入到U中

import java.util.HashSet;
import java.util.Iterator;
import java.util.Scanner;
/*
 *设G=(V,E)是无相连通带权图,找出最小生成树,也就是找出权值和最小的生成树。
 *
 测试用例:
10 6
1 2 6 1 3 1 1 4 5
2 5 3 2 3 5
3 5 6 3 6 4
3 4 5
4 6 2
5 6 6
 */
import java.util.Set;

public class Test {
    static int edge,node;   //边数、顶点数
    static int [][] c=new int[20][20];
    static boolean  []s=new boolean[20];  //如果s[i]=true,说明该顶点已经属于最小生成树的一部分
    static Scanner sc=new Scanner(System.in);
    static Set<Integer> set=new HashSet<Integer>();   //set表示U集合
    public static void main(String[] args) {
        int edge,node,f,t,v;
        edge=sc.nextInt();
        node=sc.nextInt();
        set.clear();
        for(int i=1;i<=node;i++){
            for(int j=1;j<=node;j++){
                c[i][j]=10000;
                c[j][i]=10000;
            }
            
        }
        for(int i=1;i<=edge;i++){
            f=sc.nextInt();
            t=sc.nextInt();
            v=sc.nextInt();
            c[f][t]=v;
            c[t][f]=v;
        }
        set.add(1);
        prim(node, 1);
        
    }
    
    public static void prim(int n,int u){
        s[u]=true;
        int u0,mini=1000,k=0;
        while(set.size()<n){
            Iterator<Integer> iterator = set.iterator();
            mini=1000;
            while(iterator.hasNext()){
             u0= (int)iterator.next();
                for(int i=1;i<=n;i++){
                    if(i!=u0&&c[i][u0]<mini&&s[i]==false){
                        mini=c[i][u0];
                        k=i;
                    }
                }
            }
            set.add(k);
            s[k]=true;
            System.out.println(k);
        }
    }
}

 

posted @ 2015-12-09 21:45  咸咸的告别  阅读(358)  评论(1编辑  收藏  举报