Update

problems/kamacoder/0053.寻宝-prim.md

@@ -520,8 +520,140 @@ int main() {
 
 ### Java 
 
+```Java 
+import java.util.*;
+
+public class Main {
+    public static void main(String[] args) {
+        Scanner scanner = new Scanner(System.in);
+        int v = scanner.nextInt();
+        int e = scanner.nextInt();
+
+        // 初始化邻接矩阵，所有值初始化为一个大值，表示无穷大
+        int[][] grid = new int[v + 1][v + 1];
+        for (int i = 0; i <= v; i++) {
+            Arrays.fill(grid[i], 10001);
+        }
+
+        // 读取边的信息并填充邻接矩阵
+        for (int i = 0; i < e; i++) {
+            int x = scanner.nextInt();
+            int y = scanner.nextInt();
+            int k = scanner.nextInt();
+            grid[x][y] = k;
+            grid[y][x] = k;
+        }
+
+        // 所有节点到最小生成树的最小距离
+        int[] minDist = new int[v + 1];
+        Arrays.fill(minDist, 10001);
+
+        // 记录节点是否在树里
+        boolean[] isInTree = new boolean[v + 1];
+
+        // Prim算法主循环
+        for (int i = 1; i < v; i++) {
+            int cur = -1;
+            int minVal = Integer.MAX_VALUE;
+
+            // 选择距离生成树最近的节点
+            for (int j = 1; j <= v; j++) {
+                if (!isInTree[j] && minDist[j] < minVal) {
+                    minVal = minDist[j];
+                    cur = j;
+                }
+            }
+
+            // 将最近的节点加入生成树
+            isInTree[cur] = true;
+
+            // 更新非生成树节点到生成树的距离
+            for (int j = 1; j <= v; j++) {
+                if (!isInTree[j] && grid[cur][j] < minDist[j]) {
+                    minDist[j] = grid[cur][j];
+                }
+            }
+        }
+
+        // 统计结果
+        int result = 0;
+        for (int i = 2; i <= v; i++) {
+            result += minDist[i];
+        }
+        System.out.println(result);
+        scanner.close();
+    }
+}
+
+```
+
 ### Python
 
+```python 
+def prim(v, e, edges):
+    import sys
+    import heapq
+
+    # 初始化邻接矩阵，所有值初始化为一个大值，表示无穷大
+    grid = [[10001] * (v + 1) for _ in range(v + 1)]
+
+    # 读取边的信息并填充邻接矩阵
+    for edge in edges:
+        x, y, k = edge
+        grid[x][y] = k
+        grid[y][x] = k
+
+    # 所有节点到最小生成树的最小距离
+    minDist = [10001] * (v + 1)
+
+    # 记录节点是否在树里
+    isInTree = [False] * (v + 1)
+
+    # Prim算法主循环
+    for i in range(1, v):
+        cur = -1
+        minVal = sys.maxsize
+
+        # 选择距离生成树最近的节点
+        for j in range(1, v + 1):
+            if not isInTree[j] and minDist[j] < minVal:
+                minVal = minDist[j]
+                cur = j
+
+        # 将最近的节点加入生成树
+        isInTree[cur] = True
+
+        # 更新非生成树节点到生成树的距离
+        for j in range(1, v + 1):
+            if not isInTree[j] and grid[cur][j] < minDist[j]:
+                minDist[j] = grid[cur][j]
+
+    # 统计结果
+    result = sum(minDist[2:v+1])
+    return result
+
+if __name__ == "__main__":
+    import sys
+    input = sys.stdin.read
+    data = input().split()
+    
+    v = int(data[0])
+    e = int(data[1])
+    
+    edges = []
+    index = 2
+    for _ in range(e):
+        x = int(data[index])
+        y = int(data[index + 1])
+        k = int(data[index + 2])
+        edges.append((x, y, k))
+        index += 3
+
+    result = prim(v, e, edges)
+    print(result)
+
+```
+
 ### Go
 
 ### Rust