[BOJ][⚪1][백준#03184] 양

문제 출처

BAEKJOON Online Judge #3184

문제

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

입력

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

출력

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

예제

예제 1

입력

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

출력

0 2

예제 2

입력

8 8
.######.
#..o...#
#.####.#
#.#v.#.#
#.#.o#o#
#o.##..#
#.v..v.#
.######.

출력

3 1

예제 3

입력

9 12
.###.#####..
#.oo#...#v#.
#..o#.#.#.#.
#..##o#...#.
#.#v#o###.#.
#..#v#....#.
#...v#v####.
.####.#vv.o#
.......####.

출력

3 5

My Sol

import sys
input = sys.stdin.readline

def SM(mat):
    for l in mat:
        print(*l)
    print()

def bfs(i, j):
    global co, cv, I, J, mat, visit
    Q = set()
    Q.add((i, j))
    while Q:
        ni, nj = Q.pop()
        if visit[ni][nj]: continue
        if mat[ni][nj]=='o':
            co += 1
        elif mat[ni][nj]=='v':
            cv += 1
        visit[ni][nj] = 1
        for di, dj in ((-1,0),(1,0),(0,1),(0,-1)):
            si, sj = ni+di, nj+dj
            if not (0<=si<I and 0<=sj<J): continue
            if mat[si][sj] == '#': continue
            Q.add((si, sj))

I, J = map(int, input().split())
mat = [list(input()) for _ in range(I)]
visit = [[0]*J for _ in range(I)]
to, tv = 0, 0
for i in range(I):
    for j in range(J):
        if mat[i][j]=='#': continue
        if visit[i][j]: continue
        co, cv = 0, 0
        bfs(i, j)

        if co > cv:
            to += co
        else:
            tv += cv

print(to, tv)

완전 탐색과 BFS를 사용했다. BFS 대신 DFS를 사용해도 된다.

접근 방식

1. 전체 양과 늑대의 수 to, tv
2. 마당 정보 2차원 배열 mat과 같은 크기의 visit 2차원 배열을 제작
3. 모든 2차원 배열의 각 지점에 대해 벽이 아니고 방문하지 않은 지점에서 BFS 실시
4. 같은 영역 내의 양, 늑대의 수 co, cv를 카운트하며 올려줌
5. BFS 종료 후 비교해서 전체 수에 반영

결과

맞았습니다!!