[백준] 1012번 유기농 배추

정기홍·2024년 11월 26일

코딩테스트_파이썬

목록 보기

16/49

문제

차세대 영농인 한나는 강원도 고랭지에서 유기농 배추를 재배하기로 하였다. 농약을 쓰지 않고 배추를 재배하려면 배추를 해충으로부터 보호하는 것이 중요하기 때문에, 한나는 해충 방지에 효과적인 배추흰지렁이를 구입하기로 결심한다. 이 지렁이는 배추근처에 서식하며 해충을 잡아 먹음으로써 배추를 보호한다. 특히, 어떤 배추에 배추흰지렁이가 한 마리라도 살고 있으면 이 지렁이는 인접한 다른 배추로 이동할 수 있어, 그 배추들 역시 해충으로부터 보호받을 수 있다. 한 배추의 상하좌우 네 방향에 다른 배추가 위치한 경우에 서로 인접해있는 것이다.

한나가 배추를 재배하는 땅은 고르지 못해서 배추를 군데군데 심어 놓았다. 배추들이 모여있는 곳에는 배추흰지렁이가 한 마리만 있으면 되므로 서로 인접해있는 배추들이 몇 군데에 퍼져있는지 조사하면 총 몇 마리의 지렁이가 필요한지 알 수 있다. 예를 들어 배추밭이 아래와 같이 구성되어 있으면 최소 5마리의 배추흰지렁이가 필요하다. 0은 배추가 심어져 있지 않은 땅이고, 1은 배추가 심어져 있는 땅을 나타낸다.

입력

입력의 첫 줄에는 테스트 케이스의 개수 T가 주어진다. 그 다음 줄부터 각각의 테스트 케이스에 대해 첫째 줄에는 배추를 심은 배추밭의 가로길이 M(1 ≤ M ≤ 50)과 세로길이 N(1 ≤ N ≤ 50), 그리고 배추가 심어져 있는 위치의 개수 K(1 ≤ K ≤ 2500)이 주어진다. 그 다음 K줄에는 배추의 위치 X(0 ≤ X ≤ M-1), Y(0 ≤ Y ≤ N-1)가 주어진다. 두 배추의 위치가 같은 경우는 없다.

출력

각 테스트 케이스에 대해 필요한 최소의 배추흰지렁이 마리 수를 출력한다.

DFS 풀이

import sys
# recursion 에러 대처 방안(python이 정한 최대 깊이를 변경함. )
sys.setrecursionlimit(10**6)

# 반복 횟수 입력
repeat = int(input())
# 결과 값 받을 곳 제작
result = []
# dfs 알고리즘 제작
def dfs(x, y):
  if (x < 0 or y < 0 or x >= n or y >= m):
    return False
  else:
    if (graph[x][y] == 1):
      graph[x][y] = 0
      dfs(x - 1, y)
      dfs(x + 1, y)
      dfs(x, y + 1)
      dfs(x, y - 1)
      return True

# 반복 시작
for _ in range(repeat):
  # 가로, 세로, 벌레 갯수 입력
  m, n, k = map(int, input().split())
  # 벌레 위치 입력
  graph = [[False] * (m) for _ in range(n)]
  for _ in range(k):
    y, x = map(int, input().split())
    graph[x][y] = True
  # 횟수 세기
  count = 0
  # 모든 위치를 탐색하면서, count 세기
  for y in range(m):
    for x in range(n):
      if dfs(x,y):
        count += 1
  result.append(count)
  
for i in result:
  print(i)

후기

생각보다 크게 어려운 점은 없었지만 RecursionError가 일어나서 이 부분에 대해서 적으려고 한다.
RecursionError은 재귀와 관련된 에러로써, 보통은 python에서 정한 재귀 깊이가 내가 하려고 하는 깊이보다 얉을 때 벌어진다.
- 해결 방법은 : 런타임 에러(Recursion Error)
  여기에서 확인할 수 있다.

BFS 풀이

from collections import deque

# 반복 횟수 입력
repeat = int(input())
# 결과 값 받을 곳 제작
result = []

# 이동 위치 확인하기
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]


# bfs 알고리즘 제작
def bfs(x, y):
  queue = deque()
  queue.append((x,y))
  # 인접한 모든 배추를 방문한 것으로 간주하고 모두 false로 초기화
  graph[y][x] = False
  while queue:
    a,b = queue.popleft()
    for i in range(4):
      nx = a + dx[i]
      ny = b + dy[i]
      if (nx <0 or ny <0 or nx >=m or ny >=n):
        continue
      if graph[ny][nx]:
        queue.append((nx,ny))
        graph[ny][nx] = False
  # 한 번 사이클을 돈 것이므로 1 return
  return 1

# 반복 시작
for _ in range(repeat):
  # 가로, 세로, 벌레 갯수 입력
  m, n, k = map(int, input().split())
  # 벌레 위치 입력
  graph = [[False] * (m) for _ in range(n)]
  for _ in range(k):
    x, y = map(int, input().split())
    graph[y][x] = True
  # 횟수 세기
  count = 0
  # 모든 위치를 탐색하면서, 사이클 세기
  for x in range(m):
    for y in range(n):
      if graph[y][x]:
        count += bfs(x,y)
  result.append(count)

for i in result:
  print(i)