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으로 만들어준다
결과
둘다 비슷하게 나오는것같다.
Author And Source
이 문제에 관하여(leetcode 200 (medium)), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@kjo1130/leetcode-200-medium저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)