[BOJ/Python] 3184번 문제풀이

문제 설명

미키의 뒷마당에는 특정 수의 양이 있다. 그가 푹 잠든 사이에 배고픈 늑대는 마당에 들어와 양을 공격했다.
마당은 행과 열로 이루어진 직사각형 모양이다. 글자 '.' (점)은 빈 필드를 의미하며, 글자 '#'는 울타리를, 'o'는 양, 'v'는 늑대를 의미한다.

한 칸에서 수평, 수직만으로 이동하며 울타리를 지나지 않고 다른 칸으로 이동할 수 있다면, 두 칸은 같은 영역 안에 속해 있다고 한다. 마당에서 "탈출"할 수 있는 칸은 어떤 영역에도 속하지 않는다고 간주한다.

다행히 우리의 양은 늑대에게 싸움을 걸 수 있고 영역 안의 양의 수가 늑대의 수보다 많다면 이기고, 늑대를 우리에서 쫓아낸다. 그렇지 않다면 늑대가 그 지역 안의 모든 양을 먹는다. 맨 처음 모든 양과 늑대는 마당 안 영역에 존재한다. 아침이 도달했을 때 살아남은 양과 늑대의 수를 출력하는 프로그램을 작성하라.

입력

첫 줄에는 두 정수 R과 C가 주어지며(3 ≤ R, C ≤ 250), 각 수는 마당의 행과 열의 수를 의미한다.
다음 R개의 줄은 C개의 글자를 가진다. 이들은 마당의 구조(울타리, 양, 늑대의 위치)를 의미한다.

출력

하나의 줄에 아침까지 살아있는 양과 늑대의 수를 의미하는 두 정수를 출력한다.

---

풀이 과정

이 문제도 앞서 풀었던 문제와 마찬가지로 BFS 알고리즘을 활용해서 해결할 수 있다.

해결 과정

우선 코드 작성에 앞서, 이 문제를 어떻게 해결하면 좋을지 정리해보자.

 

울타리 내부에 양과 늑대가 여러 마리 존재할 수 있고, 만약 양이 늑대보다 많다면 늑대를 울타리 내부에서 쫓아내고, 그렇지 않은 경우에는 해당 울타리 내 양들이 모두 늑대에게 잡아막힌다.

 

따라서 각 울타리 내부를 탐색할 때 아래와 같은 동작을 수행할 수 있어야 한다.

1. 양과 늑대의 마릿수를 각각 count한다,
2. 숫자를 비교하여 승부 결과 체점 후 해당 울타리 내부의 최종 양과 늑대의 머리수를 결정한다.

해당 과정으로 모든 울타리 내부영역을 탐색한 후, 모든 양과 늑대의 마리수를 출력하면 문제를 해결할 수 있을 것 같다.

import sys
input = sys.stdin.readline

#마당의 행과 열 표시
R, C = map(int, input().split())

board = [] #마당
visit = [[0] * C for _ in range(R)] #방문 체크

#입력
for _ in range(R) :
    board.append(list(input().rstrip()))
    
#상,하,좌,우
dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]

queue = []

#BFS 알고리즘 사용
def BFS(x, y):
    global w, s
    visit[x][y] = 1
    queue.append([x, y])
    
    while queue:
        x, y = queue[0][0], queue[0][1]
        del queue[0]
        
        #늑대이면 w에 1을 추가
        if board[x][y] == "v":
            w += 1
            
        #양이면 s에 1을 추가
        elif board[x][y] == "o":
            s += 1
            
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            
            #마당 범위 안에 있고, 방문하지 않았고, #이 아니라면 BFS 탐색 진행
            if 0 <= nx < R and 0 <= ny < C and board[nx][ny] != "#" and visit[nx][ny] == 0:
                queue.append([nx, ny])
                visit[nx][ny] = 1
             
Wolf = 0 #전체 늑대의 수
Sheep = 0 #전체 양의 수

for i in range(R):
    for j in range(C):
        if board[i][j] != "#" and visit[i][j] == 0:
            w = 0
            s = 0
            BFS(i, j)
            #늑대가 양보다 크거나 같으면 양은 0마리
            if w >= s:
                s = 0
            #늑대보다 양이 더 많으면 늑대는 0마리
            else:
                w = 0
                
            #늑대와 양 더하기
            Wolf += w
            Sheep += s

print(Sheep, Wolf)

 

참고 자료
https://great-park.tistory.com/118
https://lakelouise.tistory.com/80
[백준 알고리즘] 백준 3184번 양 파이썬(Python)