백준 3187 양치기 꿍 Kotlin (dfs)

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

3187번: 양치기 꿍

문제

양치기 꿍은 맨날 늑대가 나타났다고 마을 사람들을 속였지만 이젠 더이상 마을 사람들이 속지 않는다. 화가 난 꿍은 복수심에 불타 아예 늑대들을 양들이 있는 울타리안에 마구 집어넣어 양들을 잡아먹게 했다.

하지만 양들은 보통 양들이 아니다. 같은 울타리 영역 안의 양들의 숫자가 늑대의 숫자보다 더 많을 경우 늑대가 전부 잡아먹힌다. 물론 그 외의 경우는 양이 전부 잡아먹히겠지만 말이다.

꿍은 워낙 똑똑했기 때문에 이들의 결과는 이미 알고있다. 만약 빈 공간을 '.'(점)으로 나타내고 울타리를 '#', 늑대를 'v', 양을 'k'라고 나타낸다면 여러분은 몇 마리의 양과 늑대가 살아남을지 계산할 수 있겠는가?

단, 울타리로 막히지 않은 영역에는 양과 늑대가 없으며 양과 늑대는 대각선으로 이동할 수 없다.

입력

입력의 첫 번째 줄에는 각각 영역의 세로와 가로의 길이를 나타내는 두 개의 정수 R, C (3 ≤ R, C ≤ 250)가 주어진다.

다음 각 R줄에는 C개의 문자가 주어지며 이들은 위에서 설명한 기호들이다.

출력

살아남게 되는 양과 늑대의 수를 각각 순서대로 출력한다.

알고리즘 분류

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

풀이

2022.01.04 - [알고리즘 문제 풀이/백준] - 백준 3184 양 Kotlin (bfs)

 

이전 양 문제와 동일한 문제이다.

그저 그래프를 탐색하며 v와 k를 찾으면 되는 문제이므로, dfs 혹은 bfs 모두 사용 가능하다.

이번엔 dfs로 풀었다.

그래프 전체를 탐색하면서, 칸이 #으로 시작하지 않고, 아직 방문하지 않은 구역들에대해 모두 탐색하고 양과 늑대의 수를 구해 양이 살아남는지, 늑대가 살아남는지 계산하여 최종 값에 더해주면 된다.

기존 그래프에 #이 아닌 부분을 모두 탐색하게 되는데, 방문한 곳은 #으로 바꿔줌으로써

방문 체크 배열은 따로 만들지 않고 방문 체크를 할 수 있다.

코드

val dirXY = arrayOf(arrayOf(0,1),arrayOf(0,-1),arrayOf(1,0),arrayOf(-1,0))
val br = System.`in`.bufferedReader()
var sCnt = 0
var wCnt = 0
fun dfs(r : Int, c : Int, n : Int, m : Int, graph : Array<CharArray>){
    if(graph[r][c]=='v'){
        wCnt++
    }
    else if(graph[r][c]=='k'){
        sCnt++
    }
    graph[r][c] = '#'

    for(dir in 0 until 4){
        val nr = r + dirXY[dir][0]
        val nc = c + dirXY[dir][1]
        if(nr !in 0 until n || nc !in 0 until m) continue
        if(graph[nr][nc]=='#')continue
        else if(graph[nr][nc]=='v'){
            wCnt++
        }
        else if(graph[nr][nc]=='k'){
            sCnt++
        }
        graph[nr][nc]='#'
        dfs(nr,nc,n,m,graph)
    }
}

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

    val (n,m) = br.readLine().split(' ').map{it.toInt()}
    val graph = Array(n){br.readLine().toCharArray()}
    var sheep = 0
    var wolf = 0
    for(i in 0 until n){
        for(j in 0 until m){
            if(graph[i][j]!='#'){
                sCnt=0
                wCnt=0
                dfs(i,j,n,m,graph)
                if(sCnt>wCnt)
                    sheep+=sCnt
                else
                    wolf+=wCnt
            }

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