[백준] 3184번: 양 (python)

3184번: 양

---

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

 

 

입력 :

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

 

 

출력 :

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

 

 

 

 

풀이)

각 구역을 순회하면서 양과 늑대의 수를 세어 카운팅 하는 방식을 사용하면 될 것이라고 생각했다.

방문 처리를 위해 DFS를 사용했다.

 

# 울타리가 아닌 각 구역을 순회하면서
# 양과 늑대의 수를 세어 카운팅하는 방식을 취하면 될 것 같음
# 방문처리를 위해 DFS를 쓰면 될 것 같다.
 
def DFS(i, j):
    global visited, answer_sheep, answer_wolf
 
    di = [-1, 1, 0, 0] # 상하 좌우
    dj = [0, 0, -1, 1]
 
    stack = [[i, j]]
 
    sheep = 0
    wolf = 0
 
    # 해당 구역을 순회하는 DFS
    while stack:
        a, b = stack.pop()
        for i in range(4):
            x = a + di[i]
            y = b + dj[i]
            # 구역을 돌기 위한 조건
            # 뒷마당 내에서, 울타리 안
            if -1 < x < r and -1 < y < c and g[x][y] != "#":
                # 빈 필드의 경우
                if visited[x][y] == 0 and g[x][y] == ".":
                   visited[x][y] = 1
                   stack.append([x, y])
                
                # 양이 있는 경우
                elif visited[x][y] == 0 and g[x][y] == "o":
                    visited[x][y] = 1
                    sheep += 1
                    stack.append([x, y])
                
                # 늑대가 있는 경우
                elif visited[x][y] == 0 and g[x][y] == "v":
                    visited[x][y] = 1
                    wolf += 1
                    stack.append([x, y])
 
    # 해당 구역을 모두 순회했을 때, 늑대가 많으면 늑대의 수만 추가
    # 양의 수가 많으면 양의 수만 답에 추가.
    if wolf < sheep:
        answer_sheep += sheep
    else:
        answer_wolf += wolf
 
r, c = map(int, input().split())
g = [list(input()) for i in range(r)]
visited = [[False] * c for i in range(r)]
 
answer_sheep, answer_wolf = 0, 0
for i in range(r):
    for j in range(c):
        #방문하지 않은 구역인 경우
        if visited[i][j] == False:
            DFS(i, j)
 
print(answer_sheep, answer_wolf)

 

이제 실버 2,3 정도의 DFS, BFS 문제는 어느정도 감이 익는 것 같다.

여기에 여러가지 조건이 붙으면 힘들긴 하지만... ㅋㅋㅋㅋ

얼른 도가 터서 골드문제도 잘 풀었으면 좋겠다.

---

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

3184번: 양