java代码实现图论算法中的prime算法

以下是 Java 代码实现 Prime 算法的示例：

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 表示边的权值。输出最小生成树的权值和。