14年12月CCF真题4 最优灌溉

问题描述

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

输入格式

　　输入的第一行包含两个正整数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。

    生成最小树的问题，采用并查集的方式，在CCF练习系统中运行超时，得分70，代码如下：

import java.util.Arrays;
import java.util.Scanner;

public class Main {
	
	public static class Edge implements Comparable<Edge> {

		public int a,b;
		public int cost;
		@Override
		public int compareTo(Edge o) {
			if(cost > o.cost){
				return 1;
			} 
			else if (cost == o.cost){
				return 0;
			}
			return -1;
		}
	}
	
	public static final int N = 1001;
	
	static int [] tree = new int[N];
	
	static {
		for (int i = 0; i < N; i ++){
			tree[i] = -1;
		}
	}
	
	static int findRoot(int x){
		if(tree[x] == -1) return x;
		else {
			int temp = findRoot(tree[x]);
			tree[x] = temp;
			return temp;
		}
	}

	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();
		int m = scanner.nextInt();
		Edge [] edges = new Edge[m];
		for(int i =0 ;i <m; i ++){
			Edge edge = new Edge();
			edge.a = scanner.nextInt();
			edge.b = scanner.nextInt();
			edge.cost = scanner.nextInt();
			edges[i] = edge;
		}
		Arrays.sort(edges);
		int ans = 0;
		for(int i = 0;i < m;i ++){
			int a = findRoot(edges[i].a);
			int b = findRoot(edges[i].b);
			if(a != b){
				tree[a] = b;
				ans += edges[i].cost;
			}
		}
		System.out.println(ans);
	}
}

转载于:https://my.oschina.net/huding/blog/547843