[파이썬] 백준 3184번: 양

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

3184번: 양

 

문제

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

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

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

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

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

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

입력

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

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

출력

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

예제 입력 1 복사

6 6
...#..
.##v#.
#v.#.#
#.o#.#
.###.#
...###

예제 출력 1 복사

0 2

 

## BFS 문제
import sys
from collections import deque
input=sys.stdin.readline
dx=[0,0,1,-1]
dy=[1,-1,0,0]

def bfs(x,y):
    global sheep
    global wolf
    q=deque()
    q.append((x,y))
    visited[x][y]=True
    if graph[x][y]=='o':
        sheep+=1
    elif graph[x][y]=='v': # 들어온 칸도 확인해줌
        wolf+=1
    while q:
        x,y=q.popleft()
        for i in range(4):
            nx=x+dx[i]
            ny=y+dy[i]
            if 0<=nx<N and 0<=ny<M and visited[nx][ny]==False and graph[nx][ny]!='#':
                if graph[nx][ny]=='o': # 양 칸
                    visited[nx][ny]=True
                    sheep+=1
                    q.append((nx,ny))
                elif graph[nx][ny]=='v': # 늑대 칸
                    visited[nx][ny]=True
                    wolf+=1
                    q.append((nx,ny))
                elif graph[nx][ny]=='.': # 빈 칸
                    visited[nx][ny]=True
                    q.append((nx,ny))

N,M=map(int,input().split())
graph=[list(input()) for _ in range(N)]
visited=[[False for _ in range(M)] for _ in range(N)]
SHEEP, WOLF= 0,0
for i in range(N):
    for j in range(M):
        sheep,wolf=0,0
        if visited[i][j]==False and graph[i][j]!='#': # 방문하지않았고 울타리가 아니라면
            bfs(i,j) # bfs 수행
            if sheep>wolf: # 양이 늑대보다 많다면 늑대를 내쫓고 양은 살아남음
                SHEEP+=sheep
            elif sheep<=wolf: # 양이 늑대와 같거나 적다면 양은 죽고 늑대만 살음
                WOLF+=wolf

print(SHEEP,WOLF)