6---------prim算法的简单应用

用prim算法实现求通信花费问题，我们经常需要的只需要将求和的部分换成其他操作就行了

代码展示

package list;

import java.util.Scanner;
/**
 * 
 * @author 哑元
 *
 */

public class Test1 {
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        int totalPrice; //总价格
        int n,e;    //顶点数，边数
        System.out.println("请输入村庄数量");
        n = scan.nextInt();
        System.out.println("请输入边数量");
        e = scan.nextInt(); 
        boolean[] visited = new boolean[n]; //定义一个数组来记录该村庄是否有水
        //定义一个简单的图
        int edges[][] = new int[n][n];
        int nodes[] = new int[n];
        //初始化顶点表
        System.out.println("输入每个顶点的费用");
        for (int i = 0; i < nodes.length; i++) {
            visited[i] = false;
            nodes[i] = scan.nextInt();
        }
        System.out.println("输入边信息");//渠道费用
        for (int i = 0; i < e; i++) {
            int v,c;
            System.out.println("请分别输入顶点的顶点号(v,c),和边信息");
            v = scan.nextInt();
            c = scan.nextInt();
            edges[v][c] = scan.nextInt();
        }
        //开始分析,先筛选造价最低的井
        int min = 65535;
        int flag = 0;
        for (int i = 0; i < nodes.length; i++) {
            if(nodes[i] < min){
                min = nodes[i];
                flag = i;
            }
        }
        //将筛选得村庄造井
        totalPrice = nodes[flag];
        visited[flag] = true;
        //从该村庄出发，依次寻找与该村庄最近的村庄，求得最短路径，这里实际上用到的是最小生成树算法，选用prim算法
        int lowcost[] = new int[n];
        int v0 = flag;
        for (int i = 0; i < n; i++) {
            lowcost[i] = edges[v0][i];
        }
        for (int i = 0; i < n-1; i++) {
            min = 65535;
            for (int j = 0; j < n; j++) {
                if(!visited[j] && lowcost[j]<min){
                    min = lowcost[j];
                    flag = j;
                }
            }
            visited[flag] = true;
            int v = flag;
            if(min < nodes[v]){
                totalPrice += min;
            }else{
                totalPrice += nodes[v];
            }
            for (int j = 0; j < lowcost.length; j++) {
                if(!visited[j]&&edges[v][j]<lowcost[j]){    //修正当前最小生成树到其他边的距离
                    lowcost[j] = edges[v][j];
                }
            }
        }
        System.out.println(totalPrice);
    }

}

输入样式
请输入村庄数量
4
请输入边数量
6
输入每个顶点的费用
1
2
3
4
输入边信息
请分别输入顶点的顶点号(v,c),和边信息
0
1
1
请分别输入顶点的顶点号(v,c),和边信息
1
2
2
请分别输入顶点的顶点号(v,c),和边信息
0
2
3
请分别输入顶点的顶点号(v,c),和边信息
1
3
3
请分别输入顶点的顶点号(v,c),和边信息
2
3
5
请分别输入顶点的顶点号(v,c),和边信息
0
3
4

输出样式
7