[백준 3187번] 파이썬 - 양치기 꿍

백준 3187 - 양치기 꿍

시간 제한 1초, 메모리 제한 128 MB

# 조건

• 양치기 꿍은 맨날 늑대가 나타났다고 마을 사람들을 속였지만 이젠 더이상 마을 사람들이 속지 않는다.
• 화가 난 꿍은 복수심에 불타 아예 늑대들을 양들이 있는 울타리안에 마구 집어넣어 양들을 잡아먹게 했다.
• 하지만 양들은 보통 양들이 아니다.
• 같은 울타리 영역 안의 양들의 숫자가 늑대의 숫자보다 더 많을 경우 늑대가 전부 잡아먹힌다.
  • 물론 그 외의 경우는 양이 전부 잡아먹히겠지만 말이다.
• 꿍은 워낙 똑똑했기 때문에 이들의 결과는 이미 알고있다.
• 만약 빈 공간을 '.'(점)으로 나타내고 울타리를 '#', 늑대를 'v', 양을 'k'라고 나타낸다면 여러분은 몇 마리의 양과 늑대가 살아남을지 계산할 수 있겠는가?
• 단, 울타리로 막히지 않은 영역에는 양과 늑대가 없으며 양과 늑대는 대각선으로 이동할 수 없다.

입력

• 입력의 첫 번째 줄에는 각각 영역의 세로와 가로의 길이를 나타내는 두 개의 정수 R, C (3 ≤ R, C ≤ 250)가 주어진다.
• 다음 각 R줄에는 C개의 문자가 주어지며 이들은 위에서 설명한 기호들이다.

출력

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

# 접근 방법

• 그래프 탐색 문제이다.
• 한 구역 안에 양과 늑대의 수를 세고 더 많은 쪽을 살려두면 되는 문제이다.
• 오랜만에 알고리즘을 풀어서 그런지 조금은 복잡하게, 주어진 조건을 모두 활용하여 풀었다.
• 입력받은 배열에서 전체 양과 늑대 수를 세어주고 VISITED 배열을 생성해주었다.
• 입력 배열을 순회하며 늑대를 발견한 경우 BFS를 실행하여 같은 구역의 양과 늑대 수를 세어주고 더 적은 쪽을 전체 늑대 OR 양의 수에서 빼주었다.
• 사실 양 또는 늑대를 발견한다면 실행하여 더 많은 쪽을 더해가도 되는 문제이다.

import sys
input = sys.stdin.readline
from collections import deque

def bfs(si, sj):
    di, dj = [1, -1, 0, 0], [0, 0, 1, -1]
    q = deque()
    q.append((si, sj))
    cs, cw = 0, 1
    while q:
        ni, nj = q.popleft()
        for d in range(4):
            fi, fj = ni+di[d], nj+dj[d]
            if 0<=fi<R and 0<=fj<C and not visited[fi][fj]:
                if arr[fi][fj] == '#':
                    continue
                elif arr[fi][fj] == 'v':
                    cw += 1
                elif arr[fi][fj] == 'k':
                    cs += 1

                q.append((fi, fj))
                visited[fi][fj] = 1
    if cs > cw:
        return [1, cw]
    else:
        return [0, cs]

R, C = map(int, input().split())
cnt_sheep, cnt_wolf = 0, 0
arr = [list(input().strip()) for _ in range(R)]
for i in range(R):
    for j in range(C):
        if arr[i][j] == 'v':
            cnt_wolf +=1
        elif arr[i][j] == 'k':
            cnt_sheep += 1

visited = [[0] * C for _ in range(R)]

for i in range(R):
    for j in range(C):
        if arr[i][j] == 'v' and not visited[i][j]:
            visited[i][j] = 1
            flag = bfs(i, j)
            if flag[0] == 1:
                cnt_wolf -= flag[1]
            else:
                cnt_sheep -= flag[1]

print(cnt_sheep, cnt_wolf)