Baekjoon Online Judge 4963번 파이썬

문제 링크
https://www.acmicpc.net/problem/4963

4963번: 섬의 개수

2667번과 아주 유사한 문제다.
2667번에서는 델타값을
dx = [0, 0, -1, 1]
dy = [-1, 1, 0, 0] 으로 주었지만
이번 4963번 문제에서는 대각선 방향만 추가해주어야 한다.
그래서 델타 값을
dx = [0, 0, -1, 1, -1, -1, 1, 1]
dy = [-1, 1, 0, 0, -1, 1, -1, 1] 로 주었다.

계속 list out of range 오류가 떴었다.
이것은 내가 2차원 배열의 너비 w, 높이 h 값의 방향을 제대로 알고 있지 않아서 오류가 났던 것이었다.


정답 코드

import sys
sys.setrecursionlimit(10 ** 6)

dx = [0,0,-1,1,-1,-1,1,1]
dy = [-1,1,0,0,-1,1,-1,1]

def dfs(x,y):
  visited[x][y] = True
  for i in range(8):
    pointX = x + dx[i]
    pointY = y + dy[i]
    if 0<=pointX<h and 0<=pointY<w:
      if visited[pointX][pointY] == False and mapping[pointX][pointY] == 1:
        dfs(pointX,pointY)


while True:
  w, h = map(int, sys.stdin.readline().split())
  visited = [[False]*w for i in range(h)] 
  mapping = []
  count = 0
  
  if w == 0 and h == 0:
    break
  else:
    for i in range(h):
      mapping.append(list(map(int,sys.stdin.readline().split())))
    for i in range(h):
      for j in range(w): 
        if mapping[i][j] == 1 and visited[i][j] == False:
          dfs(i,j)
          count += 1
  print(count)