[백준_3187] 양치기 꿍

https://www.acmicpc.net/problem/3187

3187번: 양치기 꿍

문제

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

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

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

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

입력

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

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

출력

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

문제해결

전형적인 DFS,BFS 탐색 문제이다. 둘 모두 사용가능한 문제이며 기초로직연습하기 좋은 문제이다.

해당문제는 BFS 너비우선탐색으로 global 변수를 받아 총 늑대와 양의 갯수를 추가하는 식으로 문제를 해결하였다.

3184번 '양' 문제와 유사하다. dfs,bfs문제 간단한 로직을 연습하고 싶은분들께 추천드리고 싶은 문제이다.

 

from collections import deque
r,c = map(int,input().split())
graph = [list(input()) for _ in range(r)]
visited = [[0]*c for _ in range(r)]
dx,dy = [1,-1,0,0],[0,0,1,-1]
def in_range(x,y):
    return 0<=x<r and 0<=y<c and graph[x][y]!='#'

total_v = 0
total_k = 0
def bfs(x,y):
    global total_v, total_k
    v,k=0,0
    q = deque()
    visited[x][y] = True
    q.append((x,y))
    while q:
        a,b = q.popleft()
        for i in range(4):
            nx = dx[i]+a
            ny = dy[i]+b
            if in_range(nx, ny) and visited[nx][ny] == 0:
                q.append((nx,ny))
                visited[nx][ny]=True
                if graph[nx][ny]=='v':
                    v+=1
                elif graph[nx][ny]=='k':
                    k+=1
    if v>=k:
        total_v += v
    else:
        total_k += k
for i in range(r):
    for j in range(c):
        bfs(i,j)
print(total_k,total_v)