백준 3184 양 Kotlin (bfs)

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

3184번: 양

문제

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

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

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

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

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

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

입력

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

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

출력

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

알고리즘 분류

• 그래프 이론
• 그래프 탐색
• 너비 우선 탐색
• 깊이 우선 탐색

풀이

bfs 혹은 dfs로 풀 수 있는 간단한 그래프 탐색 문제이고 본인은 bfs로 풀었다.

벽이 아닌 곳을 만날 때마다 bfs를 돌리고, 방문 가능한 구역을 모두 방문함으로써 구역을 나누는 건 visited 배열로(방문 체크) 쉽게 구현할 수 있다. 여기서 구역마다 o의 수와 v의 수를 구하고, o>v라면 v는 0이 되고(늑대가 죽는다), o<=v라면 o는 0이 된다(양이 죽는다). 이렇게 구역마다 살아남은 o와 v의 수를 합산하여 출력하면 된다.

 

코드

import java.util.*
val dirXY = arrayOf(arrayOf(0,1),arrayOf(0,-1),arrayOf(1,0),arrayOf(-1,0))
val br = System.`in`.bufferedReader()

//3<= r,c <=250
val visited = Array(250){BooleanArray(250)}

fun bfs(i : Int, j : Int, r : Int, c : Int, graph : Array<String>) : Pair<Int,Int>{
    var sheep = 0
    var wolf = 0
    var q : Queue<Pair<Int,Int>> = LinkedList<Pair<Int,Int>>()
    q.add(Pair(i,j))
    visited[i][j]=true
    if(graph[i][j]=='o'){
        sheep++
    }
    else if(graph[i][j]=='v'){
        wolf++
    }
    while(q.isNotEmpty()){
        val cur = q.poll()

        for(dir in 0 until 4){
            val nr = cur.first + dirXY[dir][0]
            val nc = cur.second + dirXY[dir][1]
            if(nr !in 0 until r || nc !in 0 until c) continue
            if(visited[nr][nc]) continue
            if(graph[nr][nc]=='#') continue
            else if(graph[nr][nc]=='o'){
                sheep++
            }
            else if(graph[nr][nc]=='v'){
                wolf++
            }
            visited[nr][nc]=true
            q.add(Pair(nr,nc))
        }
    }
    if(sheep > wolf){
        wolf=0
    }
    else{
        sheep=0
    }

    return Pair(sheep,wolf)
}

fun main() = with(System.out.bufferedWriter()){

    val (r,c) = br.readLine().split(' ').map{it.toInt()}
    val graph = Array(r){br.readLine()}
    var sheep = 0
    var wolf = 0
    for(i in 0 until r){
        for(j in 0 until c){
            if(visited[i][j]) continue
            if(graph[i][j]=='#') continue
            var result = bfs(i,j,r,c,graph)
            sheep+=result.first
            wolf+=result.second
        }
    }

    write("$sheep $wolf")
    close()
}