<DFS BFS> 백준 1012번. 유기농 배추

https://www.acmicpc.net/problem/1012

1012번: 유기농 배추

from collections import deque
import sys
sys.setrecursionlimit(10000)
input = sys.stdin.readline
#상하좌우
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]

def dfs(x,y):
  graph[x][y] = 0
  for i in range(4):
    nx = x + dx[i]
    ny = y + dy[i]
    if 0<= nx < h and 0<= ny < w:
      if graph[nx][ny]:
        graph[nx][ny] = 0
        dfs(nx,ny)

case = int(input())
cnt_lst = []

for _ in range(case):
  w , h, n = map(int, input().split()) #가로,세로,개수
  graph = [ [0]*w for i in range(h)]  
  cnt = 0
  for _ in range(n):
    i,j = map(int, input().split())
    graph[j][i] = 1
    
  for i in range(h):  # 세로 : 0~7
    for j in range(w): #가로 : 0~9
      if graph[i][j]:
        dfs(i,j)
        cnt+= 1
  cnt_lst.append(cnt)

print(*cnt_lst, sep = "\n")

정답이다. 처음에 틀렸는데, 배추의 위치가 x(가로), y(세로)임을 주의하지 않아 리스트 범위를 초과해 틀려버렸다.

 

나름 실버 2 본인에게 어려운 문젠데 맞춰서 기분이 좋댱