leetcode 200 (medium)

문제

https://leetcode.com/problems/number-of-islands/

BFS나 DFS로 풀 수 있는 문제이다.
공간 메모리를 줄이기 위해 별도의 기록장치 없이 grid에다가 탐색이 된 곳은 값을 0으로 바꿔주면서 탐색을 하려고 한다.

풀이1 (DFS)

class Solution:
    def numIslands(self, grid: List[List[str]]) -> int:
        dx = [1,0,-1,0]
        dy = [0,-1,0,1]
        def dfs(y,x):
            if y <0 or y >= len(grid) or x <0 or x >= len(grid[0]) or grid[y][x] == '0':
                return
            
            grid[y][x] = '0'
            
            for i in range(4):
                dfs(y + dy[i],x + dx[i])
                
        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

풀이2 (BFS)

class Solution:
    def numIslands(self, grid: List[List[str]]) -> int:
        dx = [1,0,-1,0]
        dy = [0,-1,0,1]

        def bfs(y,x):
            q = deque()
            q.append((y,x))
            
            while(q):
                b,a = q.popleft()
                for i in range(4):
                    ny = b + dy[i]
                    nx = a + dx[i]
                    
                    if 0 <= ny < len(grid) and 0 <= nx < len(grid[0]) and grid[ny][nx] == '1':
                        grid[ny][nx] = '0'
                        q.append((ny,nx))

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

popleft 부분만 바꿔주면 DFS가 되기도 한다.
어쨌든 순서대로 탐색하면서 탐색이 된 곳은 모두 0으로 만들어준다

결과

둘다 비슷하게 나오는것같다.

좋은 웹페이지 즐겨찾기