[BOJ] [Python] 3187번 : 양치기 꿍

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

3187번: 양치기 꿍

문제

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

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

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

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

입력

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

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

 

출력

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

 

풀이방법

방법은 BFS로 풀었습니다.

 

 . , v, k 로 이루어진 블럭을 BFS로 찾습니다. 블럭안에 v와 k의 크기의 따라  k  > v 면 k, 0 을 반환하고 k <= v 면 0, v 를 반환하여 총합을 반환합니다.

 

 

코드

from collections import deque
import sys

input = sys.stdin.readline

r, c = map(int , input().rstrip().split())

dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]
def bfs(i, j):
    w, s = 0, 0
    visited[i][j] = True
    deq = deque([(i,j)])
    while deq:
        x, y = deq.popleft()
        if matrix[x][y] == 'k':
            s += 1
        if matrix[x][y] == 'v':
            w += 1
        for k in range(4):
            nx = x + dx[k]
            ny = y + dy[k]
            if 0 <= nx < r and 0<= ny < c and visited[nx][ny] == False and matrix[nx][ny] != '#':
                deq.append((nx,ny))
                visited[nx][ny] = True
    if s > w:
        w = 0
    else:
        s = 0
    return w, s



matrix = [list(list(input().rstrip())) for _ in range(r)]
visited = [[False] * c for _ in range(r)]

total_w, total_s = 0, 0
for i in range(r):
    for j in range(c):
        if matrix[i][j] != '#' and visited[i][j] == False:
            w,s = bfs(i,j)
            total_w += w
            total_s += s
print(total_s, total_w)