3187번: 양치기 꿍
입력의 첫 번째 줄에는 각각 영역의 세로와 가로의 길이를 나타내는 두 개의 정수 R, C (3 ≤ R, C ≤ 250)가 주어진다. 다음 각 R줄에는 C개의 문자가 주어지며 이들은 위에서 설명한 기호들이다.
www.acmicpc.net
문제
양치기 꿍은 맨날 늑대가 나타났다고 마을 사람들을 속였지만 이젠 더이상 마을 사람들이 속지 않는다. 화가 난 꿍은 복수심에 불타 아예 늑대들을 양들이 있는 울타리안에 마구 집어넣어 양들을 잡아먹게 했다.
하지만 양들은 보통 양들이 아니다. 같은 울타리 영역 안의 양들의 숫자가 늑대의 숫자보다 더 많을 경우 늑대가 전부 잡아먹힌다. 물론 그 외의 경우는 양이 전부 잡아먹히겠지만 말이다.
꿍은 워낙 똑똑했기 때문에 이들의 결과는 이미 알고있다. 만약 빈 공간을 '.'(점)으로 나타내고 울타리를 '#', 늑대를 'v', 양을 'k'라고 나타낸다면 여러분은 몇 마리의 양과 늑대가 살아남을지 계산할 수 있겠는가?
단, 울타리로 막히지 않은 영역에는 양과 늑대가 없으며 양과 늑대는 대각선으로 이동할 수 없다.
입력
입력의 첫 번째 줄에는 각각 영역의 세로와 가로의 길이를 나타내는 두 개의 정수 R, C (3 ≤ R, C ≤ 250)가 주어진다.
다음 각 R줄에는 C개의 문자가 주어지며 이들은 위에서 설명한 기호들이다.
출력
살아남게 되는 양과 늑대의 수를 각각 순서대로 출력한다.
코드
from collections import deque
r, c = map(int, input().split())
farm = []
for _ in range(r):
farm.append(list(input()))
def bfs(x, y):
sheep, wolf = 0, 0
if farm[x][y] == 'v':
wolf += 1
elif farm [x][y] == 'k':
sheep += 1
q = deque([(x, y)])
farm[x][y] = '#'
dx = [0, 0, -1, 1]
dy = [-1, 1, 0, 0]
while q:
x, y = q.popleft()
for i in range(4):
nx = x + dx[i]
ny = y + dy[i]
if 0 <= nx < r and 0 <= ny < c and farm[nx][ny] != '#':
if farm[nx][ny] == 'v': # 늑대
wolf += 1
elif farm[nx][ny] == 'k': # 양
sheep += 1
q.append((nx, ny))
farm[nx][ny] = '#' # 방문한 칸 울타리로 표시
# 양이 살아남으면 0, 늑대가 살아남으면 1
return [0, sheep] if (sheep > wolf) else [1, wolf]
s, w = 0, 0
for i in range(r):
for j in range(c):
if farm[i][j] != '#':
res, num = bfs(i, j)
if res == 0:
s += num
else:
w += num
print(s, w)
BFS를 이용하여 풀이하였다.
'문제풀이 > DFS_BFS' 카테고리의 다른 글
[Python/파이썬] 백준 18404번 현명한 나이트 (0) | 2023.08.17 |
---|---|
[Python/파이썬] 백준 1520번 내리막 길 (0) | 2023.07.31 |
[Python/파이썬] 백준 4179번 불! (0) | 2023.07.16 |
[Python/파이썬] 백준 13565번 침투 (0) | 2023.07.03 |
[Python/파이썬] 백준 2412번 암벽 등반 (0) | 2023.06.27 |