| 44 | |
| 45 | // 邻接矩阵结构下的prim算法，从节点1出发得到最小生成树的权值和 |
| 46 | public static double prim(double x) { |
| 47 | // 结余值做边权，进而统计距离 |
| 48 | // 距离只和边权有关，原始的dist、cost不重要 |
| 49 | // 从1号点出发，更新到所有点的距离，也就是value数组 |
| 50 | for (int i = 1; i <= n; i++) { |
| 51 | visit[i] = false; |
| 52 | value[i] = cost[1][i] - x * dist[1][i]; |
| 53 | } |
| 54 | visit[1] = true; |
| 55 | double sum = 0; |
| 56 | // 最小生成树一定有n-1条边，所以一共有n-1轮解锁，每次解锁新的点进入最小生成树的点集 |
| 57 | for (int i = 1; i <= n - 1; i++) { |
| 58 | // 在没有解锁的点中，找到离最小生成树的点集最近的点，进行解锁 |
| 59 | double minDist = Double.MAX_VALUE; |
| 60 | int next = 0; |
| 61 | for (int j = 1; j <= n; j++) { |
| 62 | if (!visit[j] && value[j] < minDist) { |
| 63 | minDist = value[j]; |
| 64 | next = j; |
| 65 | } |
| 66 | } |
| 67 | // 最小的边进入最小生成树的边集，解锁的点进入最小生成树的点集 |
| 68 | sum += minDist; |
| 69 | visit[next] = true; |
| 70 | // 查看新的解锁点能不能拉进其他点的距离 |
| 71 | for (int j = 1; j <= n; j++) { |
| 72 | if (!visit[j] && value[j] > cost[next][j] - x * dist[next][j]) { |
| 73 | value[j] = cost[next][j] - x * dist[next][j]; |
| 74 | } |
| 75 | } |
| 76 | } |
| 77 | // 返回最小生成树的权值和 |
| 78 | return sum; |
| 79 | } |
| 80 | |
| 81 | public static void main(String[] args) throws IOException { |
| 82 | BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); |