[99클럽 코테 스터디 32일차 TIL] 무인도 여행

💡문제

https://school.programmers.co.kr/learn/courses/30/lessons/154540

문제 설명

메리는 여름을 맞아 무인도로 여행을 가기 위해 지도를 보고 있습니다. 지도에는 바다와 무인도들에 대한 정보가 표시돼 있습니다. 지도는 1 x 1크기의 사각형들로 이루어진 직사각형 격자 형태이며, 격자의 각 칸에는 'X' 또는 1에서 9 사이의 자연수가 적혀있습니다. 지도의 'X'는 바다를 나타내며, 숫자는 무인도를 나타냅니다. 이때, 상, 하, 좌, 우로 연결되는 땅들은 하나의 무인도를 이룹니다. 지도의 각 칸에 적힌 숫자는 식량을 나타내는데, 상, 하, 좌, 우로 연결되는 칸에 적힌 숫자를 모두 합한 값은 해당 무인도에서 최대 며칠동안 머물 수 있는지를 나타냅니다. 어떤 섬으로 놀러 갈지 못 정한 메리는 우선 각 섬에서 최대 며칠씩 머물 수 있는지 알아본 후 놀러갈 섬을 결정하려 합니다.

지도를 나타내는 문자열 배열 maps가 매개변수로 주어질 때, 각 섬에서 최대 며칠씩 머무를 수 있는지 배열에 오름차순으로 담아 return 하는 solution 함수를 완성해주세요. 만약 지낼 수 있는 무인도가 없다면 -1을 배열에 담아 return 해주세요.

---

제한사항

• 3 ≤ maps의 길이 ≤ 100
  • 3 ≤ maps[i]의 길이 ≤ 100
  • maps[i]는 'X' 또는 1 과 9 사이의 자연수로 이루어진 문자열입니다.
  • 지도는 직사각형 형태입니다.

---

입출력 예

maps	result
["X591X","X1X5X","X231X", "1XXX1"]	[1, 1, 27]
["XXX","XXX","XXX"]	[-1]

---

입출력 예 설명

입출력 예 #1

위 문자열은 다음과 같은 지도를 나타냅니다.

!https://user-images.githubusercontent.com/62426665/206862823-4633fbf1-c075-4d35-b577-26f504dcd332.png

연결된 땅들의 값을 합치면 다음과 같으며

!https://user-images.githubusercontent.com/62426665/209070615-ae568f20-cf06-4f88-8d4f-8e9861af2d36.png

이를 오름차순으로 정렬하면 [1, 1, 27]이 됩니다.

입출력 예 #2

위 문자열은 다음과 같은 지도를 나타냅니다.

!https://user-images.githubusercontent.com/62426665/206863265-0a371c69-d4b5-411a-972f-bdc36b90c917.png

섬이 존재하지 않기 때문에 -1을 배열에 담아 반환합니다.

 

💡키워드

• 깊이/너비 우선 탐색(DFS/BFS)

 

💡접근/해결 방법

✅ 풀이

class UnionFind:
    def __init__(self):
        self.parent = {}
        self.rank = {}
        self.size = {}

    def find(self, item):
        if item not in self.parent:
            self.parent[item] = item
            self.rank[item] = 0
            self.size[item] = 1
            return item
        if self.parent[item] != item:
            self.parent[item] = self.find(self.parent[item])
        return self.parent[item]

    def union(self, x, y):
        xroot, yroot = self.find(x), self.find(y)
        if xroot == yroot:
            return
        if self.rank[xroot] < self.rank[yroot]:
            self.parent[xroot] = yroot
            self.size[yroot] += self.size[xroot]
        elif self.rank[xroot] > self.rank[yroot]:
            self.parent[yroot] = xroot
            self.size[xroot] += self.size[yroot]
        else:
            self.parent[yroot] = xroot
            self.rank[xroot] += 1
            self.size[xroot] += self.size[yroot]

def solution(maps):
    rows, cols = len(maps), len(maps[0])
    uf = UnionFind()
    island_values = {}

    for i in range(rows):
        for j in range(cols):
            if maps[i][j] != 'X':
                current = (i, j)
                uf.find(current)
                island_values[current] = int(maps[i][j])
                
                for di, dj in [(0, 1), (1, 0)]:  # 오른쪽과 아래만 확인
                    ni, nj = i + di, j + dj
                    if 0 <= ni < rows and 0 <= nj < cols and maps[ni][nj] != 'X':
                        neighbor = (ni, nj)
                        uf.union(current, neighbor)

    group_values = {}
    for pos, value in island_values.items():
        root = uf.find(pos)
        group_values[root] = group_values.get(root, 0) + value

    result = sorted(group_values.values())
    return result if result else [-1]