JAVA中Prime算法的原理和实现

Prime算法:是一种穷举查找算法来从一个连通图中构造一棵最小生成树。利用始终找到与当前树中节点权重最小的边,找到节点,加到最小生成树的节点集合中,直至所有节点都包括其中,这样就构成了一棵最小生成树。prime在算法中属于贪心算法的一种,贪心算法还有:KruskalDijkstra以及哈夫曼树及编码算法。

下面具体讲一下prime算法:

1、首先需要构造一颗最小生成树,以及两个节点之间的权重数组,在此我们用一个二维数组来代表这样一个连通图的形式。节点就是0~数组长度-1,10000代表节点本身,权重 >= 100代表两个节点不连通,反之连通。
构建连通图代码如下:

    // 初始化连通图
    public static void initGraph(int[][] graph, ArrayList<Integer> points) {
        for(int i = 0 ; i < graph.length; i++) {
            points.add(i);
            for(int j = 0; j < graph[i].length; j++) {
                if(i == j) {
                    graph[i][j] = 10000;
                }else {
                    int temp = (int)(Math.random() * 200 +1);
                    graph[i][j] = temp; // 大于等于100不连通, 小于100连通
                }
                graph[j][i] = graph[i][j];
            }
        }
    }

连通图的数组表示:

控制台打印数组

2、找到距离当前树中节点权重最小的边,开始节点随机产生,(算法的重点)!!!

    // prime算法实现
    public static int prime(int[][] graph, ArrayList<Integer> points, int current) {
        String path = "";
        ArrayList<Integer> selectPoints = new ArrayList<Integer>();  // 选中的点集合
        int totalWeights = 0;   //  权重总和
        selectPoints.add(current);  // 添加初始开始节点
        points.remove(current); // 从未选择的节点集合中删除被选中的节点
        path = "|" + current + "|";
        System.out.println("当前路径:" + path);
        System.out.println("当前已选中节点: " + selectPoints.toString());
        System.out.println("当前剩余节点: " + points.toString());
        System.out.println("当前总权重: " &
  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是 Java 代码实现 Prime 算法的示例: ```java import java.util.*; public class PrimeAlgorithm { private static final int INF = Integer.MAX_VALUE; public static void main(String[] args) { Scanner sc = new Scanner(System.in); int n = sc.nextInt(); // 顶点数 int m = sc.nextInt(); // 边数 int[][] graph = new int[n][n]; // 图的邻接矩阵表示 for (int i = 0; i < n; i++) { Arrays.fill(graph[i], INF); // 初始化为无穷大 } for (int i = 0; i < m; i++) { int u = sc.nextInt() - 1; // 节点编号从 0 开始 int v = sc.nextInt() - 1; int w = sc.nextInt(); graph[u][v] = graph[v][u] = w; // 无向图,需要将 u->v 和 v->u 的权值都设置 } int[] dist = new int[n]; // 记录每个节点到已连通部分的最小距离 boolean[] visited = new boolean[n]; // 记录节点是否已经加入已连通部分 Arrays.fill(dist, INF); // 初始距离为无穷大 dist[0] = 0; // 从 0 号节点开始 int ans = 0; // 最小生成树的权值和 for (int i = 0; i < n; i++) { int u = -1; for (int j = 0; j < n; j++) { if (!visited[j] && (u == -1 || dist[j] < dist[u])) { u = j; } } visited[u] = true; // 将节点 u 加入已连通部分 ans += dist[u]; // 更新最小生成树的权值和 for (int v = 0; v < n; v++) { if (!visited[v] && graph[u][v] != INF && graph[u][v] < dist[v]) { dist[v] = graph[u][v]; // 更新节点 v 到已连通部分的最小距离 } } } System.out.println(ans); } } ``` 其,输入格式为: ``` n m u1 v1 w1 u2 v2 w2 ... ``` 表示图有 n 个节点,m 条边,接下来 m 行每行表示一条边,其 ui、vi 表示边的两个端点,wi 表示边的权值。输出最小生成树的权值和。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值