[Baekjoon] 3184_양

Silver II

문제(출처: https://www.acmicpc.net/problem/3184)

---

<양>

문제 

 

미키의 뒷마당에는 특정 수의 양이 있다. 그가 푹 잠든 사이에 배고픈 늑대는 마당에 들어와 양을 공격했다.
마당은 행과 열로 이루어진 직사각형 모양이다. 글자 '.' (점)은 빈 필드를 의미하며, 글자 '#'는 울타리를, 'o'는 양, 'v'는 늑대를 의미한다.
한 칸에서 수평, 수직만으로 이동하며 울타리를 지나지 않고 다른 칸으로 이동할 수 있다면, 두 칸은 같은 영역 안에 속해 있다고 한다. 마당에서 "탈출"할 수 있는 칸은 어떤 영역에도 속하지 않는다고 간주한다.
다행히 우리의 양은 늑대에게 싸움을 걸 수 있고 영역 안의 양의 수가 늑대의 수보다 많다면 이기고, 늑대를 우리에서 쫓아낸다. 그렇지 않다면 늑대가 그 지역 안의 모든 양을 먹는다.
맨 처음 모든 양과 늑대는 마당 안 영역에 존재한다.
아침이 도달했을 때 살아남은 양과 늑대의 수를 출력하는 프로그램을 작성하라.

 

 

입력

 

첫 줄에는 두 정수 R과 C가 주어지며(3 ≤ R, C ≤ 250), 각 수는 마당의 행과 열의 수를 의미한다.
다음 R개의 줄은 C개의 글자를 가진다. 이들은 마당의 구조(울타리, 양, 늑대의 위치)를 의미한다.

 

출력 

 

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

 

 

문제 풀이

- my solution

import sys
from collections import deque

def survival(arr, visited, start, result):
    queue=deque([start])
    sheep=0;wolf=0

    if arr[start[0]][start[1]]=='v': # 늑대
        wolf+=1
    elif arr[start[0]][start[1]]=='o': # 양
        sheep+=1

    dx=[-1,1,0,0] # 상하좌우
    dy=[0,0,-1,1]

    while queue:
        temp=queue.popleft()
        for i in range(4):
            x=temp[0]+dx[i]
            y=temp[1]+dy[i]
            if 0<=x<len(arr) and 0<=y<len(arr[0]) and visited[x][y]==0:
                if arr[x][y]!="#":
                    queue.append([x,y])
                    visited[x][y]=1
                    if arr[x][y]=='v':
                        wolf+=1
                    elif arr[x][y]=='o':
                        sheep+=1
                else:
                    visited[x][y]=1

    if sheep>wolf:
        result[0]+=sheep
    else:
        result[1]+=wolf


def main():
    r,c=map(int, input().split()) # 행, 열

    arr=[]
    for i in range(r):
        arr.append(list(sys.stdin.readline().strip()))

    visited=[[0]*c for i in range(r)] # 방문 여부
    result=[0,0] # 양, 늑대 수
    for i in range(r):
        for j in range(c):
            if arr[i][j]!='#' and visited[i][j]==0: # 울타리가 아니고 아직 방문하지 않았다면
                visited[i][j]=1 # 방문
                survival(arr, visited, [i,j], result)

    print(str(result[0])+" "+ str(result[1]))


if __name__=='__main__':
    main()

 

• survival ( arr, 방문 여부, 시작 좌표, 결과 )
  : sheep, wolf  = 같은 영역에서 살아남은 양과 늑대의 수
  : queue가 빌 때까지 반복 -> 제일 처음 값을 pop 하여 상하좌우를 탐색 -> 범위 안이며 아직 방문하지 않았다면 -> 울타리가 아니라면 queue에 추가 및 방문, 양이나 늑대인 경우 wolf, sheep에 +1 but  울타리라면 방문만 표시
  : 같은 영역 탐색을 끝낸 후 양이 늑대보다 많다면 result [0]에 양의 수 더해주고 늑대가 많거나 같다면 result [1]에 늑대 수 더해줌
  
  
• main
  : 행 r, 열 c 입력
  : arr 입력
  : visited = 방문 여부
  : result = [양, 늑대]
  : arr 전체를 탐색하여 울타리가 아니고 아직 방문하지 않았다면 방문 후 함수 호출
  : result 출력

---

생각🤔

 

---