ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

最小生成树【克鲁斯卡尔】【并查集】

2022-04-09 01:02:53  阅读:160  来源: 互联网

标签:java int 查集 克鲁斯 parents 卡尔 new import public


package lanqiao;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

public class MinTree {
	static int[] parents;
	static int[] ranks;
	public static void main(String[] args) throws IOException {
		BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
		String[] s = reader.readLine().split(" ");
		int amtNode = Integer.parseInt(s[0]);
		parents = new int[amtNode+1];
		Arrays.fill(parents, -1);
		ranks = new int[amtNode+1];
		Arrays.fill(ranks, 0);
		int amtEdges = Integer.parseInt(s[1]);
		List<node> list = new ArrayList<>();
		for(int i=0;i<amtEdges;i++) {
			String[] t = reader.readLine().split(" ");
			int u = Integer.parseInt(t[0]);
			int v = Integer.parseInt(t[1]);
			int w = Integer.parseInt(t[2]);
			list.add(new node(u,v,w));
		}
		reader.close();
		Collections.sort(list);
//		System.out.println(list);
		int ans = 0;
		int ant = 0;
		for(node n:list) {
//			System.out.println(n);
			if(union(n.u,n.v)) {
				ant++;
				ans +=n.w;
			}
		}
		if(ant == amtNode-1)
		System.out.println(ans);
		else System.out.println("orz");
	}
	static int find(int x) {
		int root = x;
		while(parents[root]!=-1) {
			root = parents[root];
		}
		return root;
	}
	static boolean union(int x,int y) {
		int xRoot = find(x);
		int yRoot = find(y);
		if(xRoot == yRoot) return false;
		else {
			if(ranks[xRoot]>ranks[yRoot]) {
				parents[yRoot] = xRoot;
			}else if(ranks[xRoot]<ranks[yRoot]) {
				parents[xRoot] = yRoot;
			}else {
				parents[yRoot] = xRoot;
				ranks[yRoot]++;
			}
		}
		return true;
	}
}
class node implements Comparable<node>{
	int u;
	int v;
	int w;
	public node(int u2, int v2, int w2) {
		// TODO Auto-generated constructor stub
		this.u=u2;
		this.v=v2;
		this.w=w2;
	}
	@Override
	public int compareTo(node o) {
		// TODO Auto-generated method stub
		return Integer.compare(this.w, o.w);
	}
	@Override
	public String toString() {
		return "node [u=" + u + ", v=" + v + ", w=" + w + "]";
	}
	
}

 

标签:java,int,查集,克鲁斯,parents,卡尔,new,import,public
来源: https://www.cnblogs.com/daweiguo/p/16120601.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有