poj 1258 最小生成树（普里姆算法）

最新推荐文章于 2022-02-25 15:43:19 发布

三易的博客

最新推荐文章于 2022-02-25 15:43:19 发布

阅读量483

点赞数

CC 4.0 BY-SA版权

文章标签：算法 exception import string class

本文链接：https://blog.youkuaiyun.com/liang_love_java/article/details/7664289

本文介绍了一个使用Java实现的算法，该算法通过读取一个n×n的矩阵，并找出每一列的最小值，然后将这些最小值进行累加求和。程序包括了初始化矩阵、读取输入数据、寻找最小值并更新结果等功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import java.util.Scanner;

public class Test1258 {
	static int n = 0;
	static int[][] array = null;
	static int[] result = null;
	static boolean[] visited = null;
	static boolean b = true;

	public static void main(String[] args) throws Exception {
		Scanner scan = new Scanner(System.in);
		n = scan.nextInt();
		array = new int[n][n];
		visited = new boolean[n];
		result = new int[n];
		for (int i = 0; i < array.length; i++) {
			for (int j = 0; j < array[i].length; j++) {
				array[i][j] = scan.nextInt();
				if (b) {
					result[j] = array[i][j];
				}
			}
			b = false;
		}

		while (n > 0) {
			visited[0] = true;
			min(min(result));
			n--;
		}
		
		int sum = 0;
		for(int i=0;i<result.length; i++) {
			sum += result[i];
		}
		System.out.println(sum);
		
	}

	public static int min(int[] result) {
		int min = 0;
		int lie = 0;
		for (int i = 0; i < result.length; i++) {
			if (result[i] != 0 && visited[i] == false) {
				min = result[i];
				lie = i;
				break;
			}
		}

		for (int i = 0; i < result.length; i++) {
			if (result[i] != 0 && visited[i] == false) {
				if (min > result[i]) {
					min = result[i];
					lie = i;
				}
			}
		}
		
		if (lie != 0) {
			visited[lie] = true;
		}
		
		return lie;
	}

	public static void min(int count) {
		int row = count;
		for (int i = 1; i < result.length; i++) {
			if (visited[i] == false && array[row][i] != 0) {
				if (result[i] == 0) {
					result[i] = array[row][i];
				} else if (result[i] > array[row][i]) {
					result[i] = array[row][i];
				}

			}
		}
	}

}