[BOJ #3187] 양치기 꿍

문제

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

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

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

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

 

입력

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

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

 

출력

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

 

내가 쓴 코드

import sys; input = sys.stdin.readline
sys.setrecursionlimit(10**8)
di = [0, 0, 1, -1]
dj = [1, -1, 0, 0]

def dfs(a, b):
    global s_cnt, w_cnt
    for k in range(4):
        ni = a + di[k]
        nj = b + dj[k]
        if 0<=ni<R and 0<=nj<C and not visited[ni][nj] and arr[ni][nj] != '#':
            # 4방향 중 '#'이 나오면 중단, '.'이면 계속 진행해야 함.(visited, dfs())
            visited[ni][nj] = True
            if arr[ni][nj] == 'v':
                w_cnt += 1
            elif arr[ni][nj] == 'k':
                s_cnt += 1
            dfs(ni, nj)

R, C = map(int, input().split())
arr = [list(input()) for _ in range(R)]
visited = [[False] * C for _ in range(R)]
sheep = 0
wolf = 0
for i in range(R):
    for j in range(C):
        s_cnt = 0
        w_cnt = 0
        if arr[i][j] == 'v' and not visited[i][j]:
            visited[i][j] = True
            w_cnt += 1
            dfs(i, j)
        elif arr[i][j] == 'k' and not visited[i][j]:
            visited[i][j] = True
            s_cnt += 1
            dfs(i, j)

        if s_cnt > w_cnt:
            sheep += s_cnt
        else:
            wolf += w_cnt

print(sheep, wolf)

처음에 풀면서 아니 이럴거면 울타리 왜있는거지? 했는데 dfs() 안에서 4방향 중에 울타리 발견되면 멈춰야 하는 것이었다. ㅋㅋㅋ 그리고 v, k 일때 말고 . 일때도 visit 처리, dfs(ni, nj)를 해주어야 한다. dfs 함수 내에서 조금 버벅 거렸지만, 그래도 많이 발전했다 :D ~

 

제출 결과