(Python/🥈1)백준리즘알고리즘 3187번: 양치기 꿍

 

문제 바로가기

양치기 꿍

문제:

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

입력:

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

출력:

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

 

풀이:

 

import sys
from collections import deque

sys.stdin = open('/Users/song/Desktop/Python/Python/h.txt', 'r')

N, M = map(int,sys.stdin.readline().split(" "))
board = [list(input().strip()) for _ in range(N)]
move = [(0,1),(0,-1),(1,0),(-1,0)]

for i in range(N):
    for j in range(M):
        if board[i][j] == ".":
            board[i][j] = 0
        elif board[i][j] == "#":
            board[i][j] = 1
Q =deque()
def bfs(x, y):
    Q.append([x, y])
    visited[x][y] = True
    v, k = 0, 0
    while Q:
        x, y = Q.popleft()
        if board[x][y] == 'v':
            v += 1
        elif board[x][y] == 'k':
            k += 1
        for dx, dy in move:
            nx = dx + x
            ny = dy + y
            if 0 <= nx < N and 0<= ny < M and board[nx][ny] != 1 and visited[nx][ny] == False :
                Q.append((nx,ny))
                visited[nx][ny] = 1
    if v >= k:
        k = 0
    else:
        v = 0
    return [v,k]


V,K = 0,0
visited = [[False]*M for _ in range(N)]
for i in range(N):
    for j in range(M):
        if board[i][j] != 1 and visited[i][j] == False:
            v, k = bfs(i,j)
            V += v
            K += k
print(K,V)

 

 

기본 bfs 문제이다. 

편리를 위해 . 과 # 을 . 은 0 으로 #은 1로 바꾸어주고 하였다.