[백준 알고리즘] 백준 3184번 양 파이썬(Python)

@notepad_jj2

츄르사려고 코딩하는 코집사입니다.

---

1. [백준 알고리즘] 백준 3184번 양 파이썬(Python)

1) 문제번호 : 3184번

 

2) 문제 출처

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

3184번: 양

 

2. 문제

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

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

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

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

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

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

 

3. 제약사항

4. 입력

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

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

 

5. 출력

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

 

6. 풀이

- 풀이는 아래의 링크와 똑같다.

https://yongku.tistory.com/entry/%EB%B0%B1%EC%A4%80-%EC%95%8C%EA%B3%A0%EB%A6%AC%EC%A6%98-%EB%B0%B1%EC%A4%80-3187%EB%B2%88-%EC%96%91%EC%B9%98%EA%B8%B0-%EA%BF%8D-%ED%8C%8C%EC%9D%B4%EC%8D%ACPython

[백준 알고리즘] 백준 3187번 양치기 꿍 파이썬(Python)

 

7. 소스 코드

import sys
input=sys.stdin.readline

R, C = map(int, input().split()) # 마당의 행과 열

# 마당
board = []

# 방문 체크
visit = [[0] * C for _ in range(R)]

# 입력
for _ in range(R) :
    board.append(list(input().rstrip()))

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

queue = []
# BFS
def BFS(x,y) :
    global w, s # w는 늑대, s는 양

    # 방문 완료
    visit[x][y] = 1

    queue.append([x,y])

    while queue :
        x,y = queue[0][0], queue[0][1]
        del queue[0]

        # 늑대면 w에 1 추가
        if board[x][y] == "v":
            w += 1

        # 양이면 s에 1 추가
        elif board[x][y] == "o":
            s += 1

        # 4방 탐색
        for i in range(4) :
            nx = x + dx[i]
            ny = y + dy[i]

            # 마당 범위 안에 있고, 방문하지 않았고 #이 아니라면 BFS 돌자
            if 0<=nx<R and 0<=ny<C and board[nx][ny]!= "#" and visit[nx][ny] == 0 :
                queue.append([nx,ny])
                visit[nx][ny] = 1

Wolf = 0 # 전체 늑대의 수
Sheep = 0 # 전체 양의 수

for i in range(R) :
    for j in range(C) :
        # #이 아니고 방문하지 않았다면 BFS 돌자
        if board[i][j] != "#" and visit[i][j] == 0 :
            w = 0 # 늑대
            s = 0 # 양
            BFS(i,j)
            # 늑대가 양보다 크거나 같으면 양은 0마리
            if w>=s :
                s = 0
            # 늑대보다 양이 더 많으면 늑대는 0마리
            else :
                w = 0

            # 늑대와 양 더하기
            Wolf += w
            Sheep += s

# 출력
print(Sheep, Wolf)

 

---