알고리즘 - 섬의 개수

리트코드 20번 문제

1을 육지로, 0을 물로 가정한 그리드 맵이 주어졌을때, 섬의 개수를 계산하라.(연결되어 있는 1의 덩어리 개수를 구하라)

Input

[
[“1”,”1”,”0”,”0”,”0”],
[“1”,”1”,”0”,”0”,”0”],
[“0”,”0”,”1”,”0”,”0”],
[“0”,”0”,”0”,”1”,”1”] ]

Output: 3

Solution1 (DFS)

class Solution:
    def numIslands(self, grid: List[List[str]]) -> int:
        def dfs(i, j):
            # 더 이상 땅이 아닌 경우
            if i < 0 or i >= len(grid) or j < 0 or j >= len(grid[0]) or grid[i][j] != '1':
                return
            # 탐색 한 것은 다른 값으로 바꿈
            grid[i][j] = '#'
            print(i, j)
            dfs(i + 1, j)
            dfs(i - 1, j)
            dfs(i, j + 1)
            dfs(i, j - 1)

        count = 0
        for i in range(len(grid)):
            for j in range(len(grid[0])):
                if grid[i][j] == '1':
                    dfs(i, j)
                    count += 1
        return count

댓글남기기