hdu1223 还是畅通工程 kruskal裸题并查集入门

原创于 2019-03-19 21:36:37 发布 · 229 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#最小生成树 #并查集 #kruskal

并查集同时被 2 个专栏收录

12 篇文章

订阅专栏

最小生成树

5 篇文章

订阅专栏

博客围绕“畅通工程”问题展开，该问题要求使全省村庄间公路交通可达且公路总长度最小。给出了问题描述、输入输出格式及示例，作者用并查集解决此最小生成树问题，未使用kruskal模板。

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

还是畅通工程

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

Problem Description

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

Input

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

Output

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

Sample Input

1 2 1

1 3 2

2 3 4

1 2 1

1 3 4

1 4 1

2 3 3

2 4 2

3 4 5

Sample Output

Hint

Hint Huge input, scanf is recommended.

第一道最小生成树题，kruskal的模板我还没看，直接并查集撸了，hhh，个人感觉像纯并查集_模板题，弄个Edge结构体，排序，并查集连接，贪心

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

public class hdu1233 {

	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		while(in.hasNext()) {
			n = in.nextInt();
			if(n==0)
				break;
			m = n*(n-1)/2;
			ans = 0;
			cnt = 0;
			f = new int[n+5];
			g = new Edge[m+5];
			for(int i=1;i<=n;i++)
				f[i] = i;
			for(int i=1;i<=m;i++)
				g[i] = new Edge(in.nextInt(), in.nextInt(), in.nextInt());
			Arrays.sort(g,1,m+1);
			for(int i=1;i<=m;i++) {
				union(g[i].a, g[i].b, g[i].c);
				if(cnt==n-1)
					break;
			}
			
			System.out.println(ans);
		}
		
	}
	
	static int n,m,ans=0,cnt=0;
	static int[] f;
	static Edge[] g;
	
	static int find(int x) {
		if(f[x]==x)
			return x;
		return f[x] = find(f[x]);
	}
	
	static void union(int x,int y,int w) {
		int a = find(x);
		int b = find(y);
		if(a!=b) {
			f[a] = b;
			ans+=w;
			cnt++;
		}
	}
	
	
	static class Edge implements Comparable<Edge>{
		int a,b,c;
		public Edge(int a,int b,int c) {
			this.a = a;
			this.b = b;
			this.c = c;
		}
		@Override
		public int compareTo(Edge o) {
			return this.c - o.c;
		}
	}
	
}