[Java] Union-Find

유니온 파인드 베이스 코드입니다.

import java.util.Arrays;

public class Main {
	static int N;
	static int[] parents;
	static int[] rank;
	
	static void make() {
		parents = new int[N];
		rank = new int[N];
		
		for(int i=0; i<N; i++) {
			parents[i] = i; // 자신의 부모님을 자신으로 함 
		}
	}
	
	static int find(int a) { // a가 속한 집합 찾기 (집합 대표자 반환) 
		if(a == parents[a]) {
			return a;
		}
		return parents[a] = find(parents[a]); // 경로 압축
	}
	
	// 랭크 관리 하는  배열 만들어라, 두 집합이 다를경우 큰쪽이 작은 쪽 붙이고 같으면 다른쪽 붙이면 된다.
	static boolean union(int a, int b) { // a가 속한 집합과 b가 속한 집합을  합침 (대장끼리 합쳐야함)
		int aRoot = find(a);
		int bRoot = find(b);
		
		if(aRoot == bRoot) {
			return false; // 동일한 조직일 때는 합치기 실패
		}
		
		if(rank[aRoot] > rank[bRoot]) {
			parents[bRoot] = aRoot;
		} else if(rank[bRoot] > rank[aRoot]) {
			parents[aRoot] = bRoot;
		} else {
			parents[bRoot] = aRoot;
			rank[aRoot]++;
		}
		
		return true;
	}
	
	public static void main(String[] args) {
		N = 5; // 0, 1, 2, 3, 4 5개 원소로 시작
		make();
		System.out.println(Arrays.toString(parents));
		union(0,1);
		System.out.println(Arrays.toString(parents));
		union(1,2);
		System.out.println(Arrays.toString(parents));
		union(3,4);
		System.out.println(Arrays.toString(parents));
	}
}