并查集

什么是并查集:

1、非常快的检查两个元素是否是一个集合。

2、两个元素所在各自的集合,请把他们合并在一起。

先说第一个功能:一个集合中的每个元素都会指向他的父元素,最后根元素指向他自己。判断两个元素是否是一个集合内就看他们的根元素是否是一个元素就可以了。这里有一个优化,就是一个元素可能指向他的父元素,他的父元素又指向另一个元素,另一个元素才指向根元素,在查找根元素的时候,分别把这链上的元素都指向根元素就可以了。

第二个功能:合并元素的功能,首先判断这两个集合的元素个数,把个数少的合并到个数多的上面。最后再把元素少的集合的根元素指向元素多的根元素上,就合并为一个集合了。

以下是代码:

public class UnionFind {
	public static void main(String[] args) {
		// TODO Auto-generated method stub

	}
	public static class Node {
		
	}
	public static class UnionFindSet {
		//节点和他的父节点
		public HashMap<Node, Node> fatherMap;
		//某一个节点所在的集合一共有几个节点
		public HashMap<Node, Integer> sizeMap;
		//初始化
		public UnionFindSet(List<Node> nodes) {
			makeSets(nodes);
		}
		private void makeSets(List<Node> nodes) {
			fatherMap = new HashMap<>();
			sizeMap = new HashMap<>();
			for(Node node : nodes) {
				//每个node自己形成节点
				fatherMap.put(node, node);
				sizeMap.put(node, 1);
			}
		}	
		//找头节点同时扁平化,把所有节点指向根节点
		private Node findHead(Node node) {
			Node father = fatherMap.get(node);
			if(father != node) {
				father = findHead(father);
			}
			fatherMap.put(node, father);
			return father;
		}
		//是否是一个集合
		public boolean isSameSet(Node a, Node b) {
			return findHead(a) == findHead(b);
		}
		//合并一个集合
		public void union(Node a, Node b) {
			if(a == null || b == null) {
				return;
			}
			Node aHead = findHead(a);
			Node bHead = findHead(b);
			if(aHead != bHead) {
				int aSetSize = sizeMap.get(aHead);
				int bSetSize = sizeMap.get(bHead);
				if(aSetSize <= bSetSize) {
					fatherMap.put(aHead, bHead);
					sizeMap.put(bHead, aSetSize + bSetSize);
				} else {
					fatherMap.put(bHead, aHead);
					sizeMap.put(aHead, aSetSize + bSetSize);
				}
			}
		}
	}
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值