BOJ 3187 - 양치기 꿍 (Python)

S1, BFS

문제 설명

양치기 꿍은 맨날 늑대가 나타났다고 마을 사람들을 속였지만 이젠 더이상 마을 사람들이 속지 않는다. 화가 난 꿍은 복수심에 불타 아예 늑대들을 양들이 있는 울타리안에 마구 집어넣어 양들을 잡아먹게 했다.

하지만 양들은 보통 양들이 아니다. 같은 울타리 영역 안의 양들의 숫자가 늑대의 숫자보다 더 많을 경우 늑대가 전부 잡아먹힌다. 물론 그 외의 경우는 양이 전부 잡아먹히겠지만 말이다.

꿍은 워낙 똑똑했기 때문에 이들의 결과는 이미 알고있다. 만약 빈 공간을 '.'(점)으로 나타내고 울타리를 '#', 늑대를 'v', 양을 'k'라고 나타낸다면 여러분은 몇 마리의 양과 늑대가 살아남을지 계산할 수 있겠는가?

단, 울타리로 막히지 않은 영역에는 양과 늑대가 없으며 양과 늑대는 대각선으로 이동할 수 없다.

입력

입력의 첫 번째 줄에는 각각 영역의 세로와 가로의 길이를 나타내는 두 개의 정수 R, C (3 ≤ R, C ≤ 250)가 주어진다.

다음 각 R줄에는 C개의 문자가 주어지며 이들은 위에서 설명한 기호들이다.

출력

살아남게 되는 양과 늑대의 수를 각각 순서대로 출력한다.

입력 예시

6 6
...#..
.##v#.
#v.#.#
#.k#.#
.###.#
...###

출력 예시

0 2

풀이

단순 BFS풀이
매 탐색마다 wolf와 sheep의 수를 세서 판단해주면 되는 문제

from collections import deque
from sys import stdin

n, m = map(int, stdin.readline().split())
graph = [[] for _ in range(n)]

for i in range(n):
    graph[i] = list(map(str, stdin.readline().rstrip()))

visited = [[0] * m for _ in range(n)]

def bfs(a, b):
    q = deque()
    q.append([a, b])
    wolf, sheep = 0, 0
    if graph[a][b] == 'v':
        wolf = 1
    elif graph[a][b] == 'k':
        sheep = 1

    visited[a][b] = 1
    dx = [-1, 1, 0, 0]
    dy = [0, 0, -1, 1]

    while q:
        x, y = q.popleft()

        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]

            if 0<=nx<n and 0<=ny<m and not visited[nx][ny] and graph[nx][ny] != '#':
                if graph[nx][ny] == '.':
                    q.append([nx, ny])
                    visited[nx][ny] = 1
                elif graph[nx][ny] == 'v':
                    q.append([nx, ny])
                    wolf += 1
                    visited[nx][ny] = 1
                elif graph[nx][ny] == 'k':
                    q.append([nx, ny])
                    sheep += 1
                    visited[nx][ny] = 1
    if wolf < sheep:
        return [sheep, 0]
    else:
        return [0, wolf]

res = [0, 0]
for i in range(n):
    for j in range(m):
        if graph[i][j] != '#' and not visited[i][j]:
            tmp = bfs(i, j)
            res[0] += tmp[0]
            res[1] += tmp[1]

print(*res)