[Silver I] 양 - 3184 (BFS)

[Silver I] 양 - 3184

문제 링크

성능 요약

메모리: 34144 KB, 시간: 172 ms

분류

너비 우선 탐색(bfs), 깊이 우선 탐색(dfs), 그래프 이론(graphs), 그래프 탐색(graph_traversal)

문제 설명

미키의 뒷마당에는 특정 수의 양이 있다. 그가 푹 잠든 사이에 배고픈 늑대는 마당에 들어와 양을 공격했다.

마당은 행과 열로 이루어진 직사각형 모양이다. 글자 '.' (점)은 빈 필드를 의미하며, 글자 '#'는 울타리를, 'o'는 양, 'v'는 늑대를 의미한다.

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

다행히 우리의 양은 늑대에게 싸움을 걸 수 있고 영역 안의 양의 수가 늑대의 수보다 많다면 이기고, 늑대를 우리에서 쫓아낸다. 그렇지 않다면 늑대가 그 지역 안의 모든 양을 먹는다.

맨 처음 모든 양과 늑대는 마당 안 영역에 존재한다.

아침이 도달했을 때 살아남은 양과 늑대의 수를 출력하는 프로그램을 작성하라.

입력

첫 줄에는 두 정수 R과 C가 주어지며(3 ≤ R, C ≤ 250), 각 수는 마당의 행과 열의 수를 의미한다.

다음 R개의 줄은 C개의 글자를 가진다. 이들은 마당의 구조(울타리, 양, 늑대의 위치)를 의미한다.

출력

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

풀이방법

BFS로 울타리 영역마다 방문 체크를 통하여 양(o)과 늑대(v)가 몇 마리인지 체크 후 양이 많으면 늑대는 0마리, 늑대가 많거나 같으면 양은 0마리가 되어 전체 양, 늑대 마리의 수(o_cnt, v_cnt)에 누적하여 더해준다.

o_cnt, v_cnt 출력하면 해결.

import sys
from collections import deque
sys.setrecursionlimit(10**9)
input = sys.stdin.readline

graph = []
R, C = map(int,input().split())
for i in range(R):
    graph.append(list(input()))
visited = [[False] * C for _ in range(R)]

dx = (1,-1,0,0)
dy = (0,0,1,-1)

v_cnt = 0
o_cnt = 0

for i in range(R):
    for j in range(C):
        if graph[i][j] != '#' and not visited[i][j]:
            o, v = 0, 0
            if graph[i][j] == 'v':
                v = 1
            elif graph[i][j] == 'o':
                o = 1                
            visited[i][j] = True
            q = deque()
            q.append((i,j))
            while q:
                x, y = q.popleft()

                for k in range(4):
                    nx = x + dx[k]
                    ny = y + dy[k]
                    if 0 <= nx < R and 0 <= ny < C and not visited[nx][ny] and graph[nx][ny] != '#':
                        if graph[nx][ny] == 'v':
                            v += 1
                        elif graph[nx][ny] == 'o':
                            o += 1
                        visited[nx][ny] = True
                        q.append((nx,ny))
            if o > v:
                v = 0
            else:
                o = 0
            v_cnt += v
            o_cnt += o
            
print(o_cnt, v_cnt)