prim 最小產生樹演算法 java實現

import java.io.BufferedInputStream;
import java.util.Scanner;



public class Prim {

static int[][] arr;
static boolean flag[];  //用來標記節點i是否被覆蓋
    static int n;
static int sum;
static final int maxInt = Integer.MAX_VALUE;
public static void main(String[] args) {
        Scanner s = new Scanner(new BufferedInputStream(System.in));
while(s.hasNextInt()){
            n = s.nextInt();
            arr = new int[n+1][n+1];
            flag = new boolean[n+1];
for(int i=1; i<=n; i++)
for(int j=1; j<=n; j++)
                    arr[i][j] = s.nextInt(); //從下標1開始儲存
            
            sum = 0;
            flag[1] = true; //選取第一個節點
            
for(int k=1; k<n; k++){    //迴圈n-1次
                int min = maxInt,min_i = 0;
for(int i=1; i<=n; i++){
if( !flag[i] && arr[1][i] < min){
                        min = arr[1][i];
                        min_i = i;
                    }
                }
                System.out.println("  min_i:"+min_i);
                flag[min_i] = true; // 覆蓋節點
                

for(int i=1; i<=n; i++){  //更新未覆蓋節點的距離,每加入一個點就更新
                    if( !flag[i] && arr[1][i] > arr[min_i][i])
                        arr[1][i] = arr[min_i][i];
                    System.out.print("arr[1]["+i+"]"+arr[1][i]+" ");
                }
                System.out.println();
                sum += arr[1][min_i];//加上權值
            }


            System.out.println(sum);

        }
    }

}