算法分析
比较裸的最小生成树问题
数据范围过小,可以使用prim
算法或者kurskal
算法,由于题目直接给了邻接矩阵,所有下面使用的是prim
算法
时间复杂度 $O(n^2)$
参考文献
算法提高课
Java 代码
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Scanner;
public class Main {
static int N = 110;
static int INF = 0x3f3f3f3f;
static int n;
static int[][] g = new int[N][N];
static int[] dist = new int[N];
static boolean[] st = new boolean[N];
static int prim()
{
Arrays.fill(dist, INF);
int res = 0;
for(int i = 0;i < n;i ++)
{
int t = -1;
for(int j = 1;j <= n;j ++)
{
if(!st[j] &&(t == -1 || dist[t] > dist[j]))
t = j;
}
if(i != 0 && dist[t] == INF) return INF;
st[t] = true;
if(i != 0) res += dist[t];
for(int j = 1;j <= n;j ++) dist[j] = Math.min(dist[j], g[t][j]);
}
return res;
}
public static void main(String[] args) {
Scanner scan = new Scanner(System.in);
n = scan.nextInt();
for(int i = 1;i <= n;i ++)
for(int j = 1;j <= n;j ++)
g[i][j] = scan.nextInt();
System.out.println(prim());
}
}