Java P3366 【模板】最小生成树

最新推荐文章于 2022-09-14 17:51:26 发布

小崔崔谁用的

最新推荐文章于 2022-09-14 17:51:26 发布

阅读量436

点赞数

分类专栏：洛谷 java 蓝桥杯文章标签： java

本文链接：https://blog.youkuaiyun.com/weixin_45665566/article/details/121621117

版权

java 同时被 3 个专栏收录

75 篇文章

订阅专栏

洛谷

63 篇文章

订阅专栏

蓝桥杯

7 篇文章

订阅专栏

这篇博客详细介绍了如何运用Kruskal算法来求解图论中的最小生成树问题。代码示例展示了如何在Java中实现该算法，包括边的排序、并查集的初始化和连接操作。通过这个算法，可以找到连接所有顶点的最短路径总和。当算法结束时，若图仅剩一个连通块，则表明成功构建了最小生成树；否则，输出失败提示。

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

洛谷题目链接

// czl
import java.util.*;
public class Main 
{
	static class Node implements Comparable<Node> {
		int x1, x2;
		int len;
		public Node(int a, int b, int c) {
			x1 = a; x2 = b; len = c;
		}
		public int compareTo(Node o) {
			return this.len - o.len; // 按边从小到大排序
		}
	}
	
	static int N = 5010, M = 200010;
	
	static int p[] = new int[N];
	static Node[] es = new Node[M]; 
	static int n, m, res;
	
	static int find(int x) {
		if(p[x] != x) p[x] = find(p[x]);
		return p[x];
	}
	
	static boolean kruskal() {
		
		Arrays.sort(es, 0, m); // 按边从小到大排序
		
		for(int i = 1; i <= n; i ++ ) p[i] = i; // 初始化并查集数组
		
		int cnt = n; // 刚开始有n个连通块
		for(int i = 0; i < m; i ++ ) {
			int a = find(es[i].x1);
			int b = find(es[i].x2);
			if(a != b) {
				p[a] = b;
				res += es[i].len;
				cnt -- ;
			}
		}
		// 最后应该只剩下一个连通块， 否则该图不连通！
		return cnt == 1;
	}
	
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        m = sc.nextInt();
        for(int i = 0; i < m; i ++ ) {
        	int a = sc.nextInt();
        	int b = sc.nextInt();
        	int dist = sc.nextInt();
        	es[i] = new Node(a, b, dist);
        }
        
        if(kruskal())
        	System.out.println(res);
        else
        	System.out.println("orz");
    }
}