CCF系列之最优灌溉(201412-4)

试题编号：201412-4
试题名称：最优灌溉
时间限制： 1.0s
内存限制： 256.0MB

问题描述

　　雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。
　　为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。
　　现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。

输入格式

　　输入的第一行包含两个正整数n, m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
　　接下来m行，每行包含三个整数a_i, b_i, c_i，表示第a_i片麦田与第b_i片麦田之间可以建立一条水渠，所需要的费用为c_i。

输出格式

　　输出一行，包含一个整数，表示灌溉所有麦田所需要的最小费用。

样例输入

4 4
1 2 1
2 3 4
2 4 2
3 4 3

样例输出

6

样例说明

　　建立以下三条水渠：麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。

评测用例规模与约定

　　前20%的评测用例满足：n≤5。
　　前40%的评测用例满足：n≤20。
　　前60%的评测用例满足：n≤100。
　　所有评测用例都满足：1≤n≤1000，1≤m≤100,000，1≤c_i≤10,000。

 

解题思路：

 

实例代码（java）:
　　

package ccf_text2014_12;

import java.util.Scanner;

public class BestGuanGai2 {
    
    static final int MAX = Integer.MAX_VALUE;

    static int primAlgorithm(int[][] graphMatrix, int vertexNumber, int initialVertex) {
        int minCost = 0;
        boolean[] visited = new boolean[vertexNumber + 1];
        for (int i = 0; i < vertexNumber + 1; i++) {
            visited[i] = false;
        }

        int position = initialVertex;
        visited[position] = true;
        int[] lowCostInLine = new int[vertexNumber + 1];
        int min = MAX;

        for (int i = 1; i < vertexNumber + 1; i++) {
            lowCostInLine[i] = graphMatrix[position][i];
        }
        for (int i = 1; i < vertexNumber; i++) {
            for (int j = 1; j < vertexNumber + 1; j++) {
                if (visited[j] == false && lowCostInLine[j] < min) {
                    min = lowCostInLine[j];
                    position = j;
                }
            }
            minCost += min;
            min = MAX;
            visited[position] = true;
            for (int j = 1; j < vertexNumber + 1; j++) {
                if (visited[j] == false && graphMatrix[position][j] < lowCostInLine[j]) {
                    lowCostInLine[j] = graphMatrix[position][j];
                }
            }
        }
        return minCost;
    }

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n, m;
        n = sc.nextInt();
        m = sc.nextInt();
        int[][] adjMatrix = new int[n + 1][n + 1];
        int result = 0;

        for (int i = 1; i < n + 1; i++) {
            for (int j = 1; j < n + 1; j++) {
                adjMatrix[i][j] = MAX;
            }
        }
        for (int i = 0; i < m; i++) {
            int begin = sc.nextInt();
            int end = sc.nextInt();
            int weight = sc.nextInt();
            adjMatrix[begin][end] = weight;
            adjMatrix[end][begin] = weight;
        }
        result = primAlgorithm(adjMatrix, n, 4);
        System.out.println(result);
    }
}

 

运行结果：

　　

 

ps:上述代码采用普里姆Prim算法生成用于解决最优路径的最小生成树，该代码在CCF认证平台上的得分与选择加入最小生成树顶点集的初始定点initialVertex以及系统运行环境有关，具体情况见图，有时可以满分100分通过，有时会因为运行超时只能得90分甚至80分，各位可以看看哪里还可以进行优化，将代码复制到CCF认证平台上进行测试时，注意其提交要求，我的程序没有使用package语句来定义包的信息。（如果定义了将无法评测），所以需要将以上代码的package语句删除