[Programmers] 섬 연결하기 - 탐욕법(Greedy)
import java.util.Arrays;
// 섬 연결하기 - 탐욕법(Greedy) + Union-Find 알고리즘
public class ConnectIsland { // Union - Find 알고리즘 적용
public int solution(int n, int[][] costs) {
int answer = 0, connection[] = new int[n];
for (int i = 0; i < n; i++) { // Union - Find (자기 자신을 부모 노드로 갖는 connection 배열로 초기화)
connection[i] = i;
}
Arrays.sort(costs, (o1, o2) -> o1[2] - o2[2]);
for (int[] cost : costs) { // Greedy 알고리즘
if (find(connection, cost[0], cost[1])) {
answer += cost[2];
union(connection, cost[0], cost[1]);
}
}
return answer;
}
private int get(int[] connection, int x) { // Union - Find (그래프 내에서 연결된 부모 노드를 찾는 함수)
if (connection[x] == x)
return x;
return get(connection, connection[x]); // Recursive
}
private void union(int[] connection, int a, int b) { // Union - Find (a, b 노드가 연결되면 connection 배열에 적용하는 함수)
a = get(connection, a);
b = get(connection, b);
if (a > b) { // a, b 중 작은값을 저장
connection[a] = b;
} else {
connection[b] = a;
}
}
private boolean find(int[] connection, int a, int b) { // Union - Find (a, b 노드가 서로 연결되어 있는지 확인하는 함수)
a = get(connection, a);
b = get(connection, b);
return a == b ? false : true; // a == b 이면 연결되어 있는 상태, a != b 이면 연결되어 있지 않은 상태
}
}
- Union-Find 알고리즘 적용
https://www.youtube.com/watch?v=AMByrd53PHM&list=PLRx0vPvlEmdDHxCvAQS1_6XV4deOwfVrz&index=18
Author And Source
이 문제에 관하여([Programmers] 섬 연결하기 - 탐욕법(Greedy)), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@kmdngmn/Programmers-섬-연결하기-탐욕법Greedy-Union-Find-알고리즘저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)