📩 출처
문제
미키의 뒷마당에는 특정 수의 양이 있다. 그가 푹 잠든 사이에 배고픈 늑대는 마당에 들어와 양을 공격했다.
마당은 행과 열로 이루어진 직사각형 모양이다. 글자 '.' (점)은 빈 필드를 의미하며, 글자 '#'는 울타리를, 'o'는 양, 'v'는 늑대를 의미한다.
한 칸에서 수평, 수직만으로 이동하며 울타리를 지나지 않고 다른 칸으로 이동할 수 있다면, 두 칸은 같은 영역 안에 속해 있다고 한다. 마당에서 "탈출"할 수 있는 칸은 어떤 영역에도 속하지 않는다고 간주한다.
다행히 우리의 양은 늑대에게 싸움을 걸 수 있고 영역 안의 양의 수가 늑대의 수보다 많다면 이기고, 늑대를 우리에서 쫓아낸다. 그렇지 않다면 늑대가 그 지역 안의 모든 양을 먹는다.
맨 처음 모든 양과 늑대는 마당 안 영역에 존재한다.
아침이 도달했을 때 살아남은 양과 늑대의 수를 출력하는 프로그램을 작성하라.
입력
첫 줄에는 두 정수 R과 C가 주어지며(3 ≤ R, C ≤ 250), 각 수는 마당의 행과 열의 수를 의미한다.
다음 R개의 줄은 C개의 글자를 가진다. 이들은 마당의 구조(울타리, 양, 늑대의 위치)를 의미한다.
출력
하나의 줄에 아침까지 살아있는 양과 늑대의 수를 의미하는 두 정수를 출력한다.
👉 생각
- 하나의 영역에서 너비우선탐색을 통해 양과 늑대의 수를 카운트한다.
- 양이 늑대보다 많다면, 늑대를 0으로, 그 반대라면 양을 0으로 출력하는 bfs 함수를 만들어 준다.
- 마당을 나타내는 이차원 배열에서 반복문을 통해 값을 확인하고 울타리거나 이전에 방문한 곳이 아니라면 새로운 영역이기 때문에 너비우선탐색을 돌리면 된다!
from collections import deque
import sys
def bfs(x,y):
q = deque([(x,y)])
visited[x][y] = 1
cnt_sheep, cnt_wolf = 0, 0
while q:
x, y = q.popleft()
if arr[x][y] == "o":
cnt_sheep += 1
elif arr[x][y] == 'v':
cnt_wolf += 1
for dx, dy in [(-1,0),(1,0),(0,-1),(0,1)]:
nx, ny = x + dx, y + dy
if 0 <= nx < r and 0 <= ny < c and arr[nx][ny] != "#" and not visited[nx][ny]:
q.append((nx, ny))
visited[nx][ny] = 1
if cnt_wolf >= cnt_sheep:
cnt_sheep = 0
else:
cnt_wolf = 0
return cnt_sheep, cnt_wolf
r, c = map(int, input().split())
arr = [list(sys.stdin.readline().rstrip()) for _ in range(r)]
visited = [[0] * c for _ in range(r)]
sheep, wolf = 0, 0
for i in range(r):
for j in range(c):
if arr[i][j] != '#' and not visited[i][j]:
s, w = bfs(i, j)
sheep += s
wolf += w
print(sheep, wolf)