HDU 1233还是畅通工程

最新推荐文章于 2020-02-10 23:38:28 发布

原创最新推荐文章于 2020-02-10 23:38:28 发布 · 264 阅读

0 ·

CC 4.0 BY-SA版权

解决政府‘畅通工程’目标，计算使全省任意两个村庄间都能间接通过公路可达的最小公路总长度。

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

还是畅通工程

Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 28085 Accepted Submission(s): 12527

Problem Description

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

Output

对每个测试用例，在1行里输出最小的公路总长度。

Sample Input

Sample Output

3
5


Hint
Hint
 
Huge input, scanf is recommended.

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

public class Main {

	public static void main(String[] args) {
		Scanner cin = new Scanner(new BufferedInputStream(System.in));

		while (cin.hasNext()) {
			int N = cin.nextInt();
			if (N == 0)
				break;

			int len = N * (N - 1) / 2 + 1;
			Node[] nodes = new Node[len];
			for (int i = 1; i < len; i++) {
				nodes[i] = new Node(cin.nextInt(), cin.nextInt(), cin.nextInt());
			}

			Arrays.sort(nodes, 1, len);

			int[] pre = new int[N + 1];
			for (int i = 1; i <= N; i++) {
				pre[i] = i;
			}

			int total = 0;
			int n = 0;
			for (int i=1; i<len; i++) {
				Node node = nodes[i];
				int sp = getGrandPa(node.source, pre);
				int tp = getGrandPa(node.target, pre);
				
				if(sp != tp) {
					pre[sp] = tp;
					total += node.distance;
					n ++;
					if(n >= N) {
						break;
					}
				}
			}
			System.out.println(total);
		}

		cin.close();
	}

	static int getGrandPa(int s, int[] pre) {
		int r = s;
		while (pre[r] != r) {
			r = pre[r];
		}
		
		int i = s, j;
		while(pre[i] != r) {
			j = pre[i];
			pre[i] = r;
			i = j;
		}
		return r;
	}

}

class Node implements Comparable<Node> {
	int source;
	int target;
	int distance;

	public Node(int source, int target, int distance) {
		this.source = source;
		this.target = target;
		this.distance = distance;
	}

	@Override
	public int compareTo(Node o) {
		return this.distance - o.distance;
	}
}