백준 2636 [kotlin]

문제

아래 <그림 1>과 같이 정사각형 칸들로 이루어진 사각형 모양의 판이 있고, 그 위에 얇은 치즈(회색으로 표시된 부분)가 놓여 있다. 판의 가장자리(<그림 1>에서 네모 칸에 X친 부분)에는 치즈가 놓여 있지 않으며 치즈에는 하나 이상의 구멍이 있을 수 있다.

이 치즈를 공기 중에 놓으면 녹게 되는데 공기와 접촉된 칸은 한 시간이 지나면 녹아 없어진다. 치즈의 구멍 속에는 공기가 없지만 구멍을 둘러싼 치즈가 녹아서 구멍이 열리면 구멍 속으로 공기가 들어가게 된다. <그림 1>의 경우, 치즈의 구멍을 둘러싼 치즈는 녹지 않고 ‘c’로 표시된 부분만 한 시간 후에 녹아 없어져서 <그림 2>와 같이 된다.

<그림 1> 원래 치즈 모양

다시 한 시간 후에는 <그림 2>에서 ‘c’로 표시된 부분이 녹아 없어져서 <그림 3>과 같이 된다.

<그림 2> 한 시간 후의 치즈 모양

<그림 3> 두 시간 후의 치즈 모양

<그림 3>은 원래 치즈의 두 시간 후 모양을 나타내고 있으며, 남은 조각들은 한 시간이 더 지나면 모두 녹아 없어진다. 그러므로 처음 치즈가 모두 녹아 없어지는 데는 세 시간이 걸린다. <그림 3>과 같이 치즈가 녹는 과정에서 여러 조각으로 나누어 질 수도 있다.

입력으로 사각형 모양의 판의 크기와 한 조각의 치즈가 판 위에 주어졌을 때, 공기 중에서 치즈가 모두 녹아 없어지는 데 걸리는 시간과 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에는 사각형 모양 판의 세로와 가로의 길이가 양의 정수로 주어진다. 세로와 가로의 길이는 최대 100이다. 판의 각 가로줄의 모양이 윗 줄부터 차례로 둘째 줄부터 마지막 줄까지 주어진다. 치즈가 없는 칸은 0, 치즈가 있는 칸은 1로 주어지며 각 숫자 사이에는 빈칸이 하나씩 있다.

출력

첫째 줄에는 치즈가 모두 녹아서 없어지는 데 걸리는 시간을 출력하고, 둘째 줄에는 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 출력한다.

생각

내부의 빈공간과 외부의 공기 접촉면을 어떻게 구분을 할까라고 생각을 했는데,

가장자리는 공기와 무조건 닿는 부분이므로 0이기에 따로 내부의 빈공간과 구분을 지어서 찾을 필요가 없다는것을 알았다.시작좌표를 0,0으로 설정을 해두고 외곽의 공기 접촉면만 bfs 탐색을 통해 찾아내었다. 

외부의 빈공간과 내부의 빈공간을 어떻게 구분 지을까 고민하는 과정에서 시간이 걸렸다.

코드

import java.util.Deque
import java.util.LinkedList
import java.util.Queue

var visited : Array<IntArray>? = null
var chz : Array<IntArray>? = null
var x = intArrayOf(0,0,1,-1)
var y = intArrayOf(1,-1,0,0)
var M : Int = 0
var N : Int = 0
fun main()
{
    var beforeCnt = 0
    var ct = 0
    var input = readLine()!!.split(" ").map {it.toInt()}
    M = input[0]
    N = input[1]
    visited = Array(M,{IntArray(N,{0})})
    chz = Array(M,{ IntArray(N,{0}) })
    repeat(M)
    {
        i ->
        input = readLine()!!.split(" ").map {it.toInt()}
        repeat(N)
        {
            j ->
            chz!![i][j] = input[j]
        }
    }

    var time = 0
    while(true)
    {
        beforeCnt = ct
        ct = bfs()
        if(ct > 0) {
            time++
        }
        else {
            println(time)
            println(beforeCnt)
            break
        }
    }
}

fun bfs() : Int
{
    val Q : Deque<IntArray> = LinkedList()
    Q.add(intArrayOf(0,0))
    visited = Array(M,{IntArray(N,{0})})
    visited!![0][0] = 1
    var count = 0
    while(Q.isNotEmpty())
    {
        val pq = Q.removeFirst()
        repeat(4)
        {
            i ->
            val nx = pq[0] + x[i]
            val ny = pq[1] + y[i]
            if (nx in 0..M-1 && ny in 0..N-1 && visited!![nx][ny] == 0){
                if (chz!![nx][ny] == 0)
                {
                    visited!![nx][ny] = 1
                    Q.add(intArrayOf(nx,ny))
                }
                else{
                    chz!![nx][ny] = 0
                    visited!![nx][ny] = 1
                    count++
                }
            }
        }
    }
    return count
}