3184 - 양
문제
미키의 뒷마당에는 특정 수의 양이 있다. 그가 푹 잠든 사이에 배고픈 늑대는 마당에 들어와 양을 공격했다.
마당은 행과 열로 이루어진 직사각형 모양이다. 글자 '.' (점)은 빈 필드를 의미하며, 글자 '#'는 울타리를, 'o'는 양, 'v'는 늑대를 의미한다.
한 칸에서 수평, 수직만으로 이동하며 울타리를 지나지 않고 다른 칸으로 이동할 수 있다면, 두 칸은 같은 영역 안에 속해 있다고 한다. 마당에서 "탈출"할 수 있는 칸은 어떤 영역에도 속하지 않는다고 간주한다.
다행히 우리의 양은 늑대에게 싸움을 걸 수 있고 영역 안의 양의 수가 늑대의 수보다 많다면 이기고, 늑대를 우리에서 쫓아낸다. 그렇지 않다면 늑대가 그 지역 안의 모든 양을 먹는다.
맨 처음 모든 양과 늑대는 마당 안 영역에 존재한다.
아침이 도달했을 때 살아남은 양과 늑대의 수를 출력하는 프로그램을 작성하라.
코드
import sys
sys.setrecursionlimit(10**6)
def dfs(x, y):
global sheep, wolf
if x < 0 or x > R-1 or y < 0 or y > C-1 or ground[x][y] == '#' or ground[x][y] == '-':
return False
if ground[x][y] != '#':
if ground[x][y] == 'v':
wolf += 1
if ground[x][y] == 'o':
sheep += 1
ground[x][y] = '-'
dfs(x+1, y)
dfs(x-1, y)
dfs(x, y+1)
dfs(x, y-1)
return True
return False
R, C = map(int, input().split())
ground = [list(input()) for _ in range(R)]
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
sheep_cnt, wolf_cnt = 0, 0
for i in range(R):
for j in range(C):
sheep, wolf = 0, 0
if dfs(i, j):
if sheep > wolf:
sheep_cnt += sheep
else:
wolf_cnt += wolf
print(sheep_cnt, wolf_cnt)한마디
DFS를 통해 영역을 구분짓고, 각 영역에 있는 양과 늑대 수를 비교하여 계산했다.
