[백준] 알고스팟 (1261)(Kotlin)

원본 문제

문제 설명

알고스팟 운영진이 모두 미로에 갇혔다. 미로는 NM 크기이며, 총 11크기의 방으로 이루어져 있다. 미로는 빈 방 또는 벽으로 이루어져 있고, 빈 방은 자유롭게 다닐 수 있지만, 벽은 부수지 않으면 이동할 수 없다.

알고스팟 운영진은 여러명이지만, 항상 모두 같은 방에 있어야 한다. 즉, 여러 명이 다른 방에 있을 수는 없다. 어떤 방에서 이동할 수 있는 방은 상하좌우로 인접한 빈 방이다. 즉, 현재 운영진이 (x, y)에 있을 때, 이동할 수 있는 방은 (x+1, y), (x, y+1), (x-1, y), (x, y-1) 이다. 단, 미로의 밖으로 이동 할 수는 없다.

벽은 평소에는 이동할 수 없지만, 알고스팟의 무기 AOJ를 이용해 벽을 부수어 버릴 수 있다. 벽을 부수면, 빈 방과 동일한 방으로 변한다.

만약 이 문제가 알고스팟에 있다면, 운영진들은 궁극의 무기 sudo를 이용해 벽을 한 번에 다 없애버릴 수 있지만, 안타깝게도 이 문제는 Baekjoon Online Judge에 수록되어 있기 때문에, sudo를 사용할 수 없다.

현재 (1, 1)에 있는 알고스팟 운영진이 (N, M)으로 이동하려면 벽을 최소 몇 개 부수어야 하는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 미로의 크기를 나타내는 가로 크기 M, 세로 크기 N (1 ≤ N, M ≤ 100)이 주어진다. 다음 N개의 줄에는 미로의 상태를 나타내는 숫자 0과 1이 주어진다. 0은 빈 방을 의미하고, 1은 벽을 의미한다.

(1, 1)과 (N, M)은 항상 뚫려있다.

출력

첫째 줄에 알고스팟 운영진이 (N, M)으로 이동하기 위해 벽을 최소 몇 개 부수어야 하는지 출력한다.

입출력 예제1

입력	출력
3 3 011 111 110	3

입출력 예제2

입력	출력
4 2 0001 1000	0

입출력 예제3

입력	출력
6 6 001111 010000 001111 110001 011010 100010	2

문제 풀이1 (Dijkstra - Adjacency List)

import java.io.BufferedReader
import java.io.InputStreamReader
import java.util.*

data class Node(val index: Int, val distance: Int): Comparable<Node> {
    override fun compareTo(other: Node): Int {
        return this.distance - other.distance
    }
}

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {

    val (x: Int, y: Int) = readLine()
        .split(" ")
        .map { it.toInt() }

    var matrix: Array<IntArray> = Array(y) { intArrayOf() }

    for (i in 0 until y) {
        val intArr = readLine()
            .map { it.toInt() - 48 }
            .toIntArray()

        matrix[i] = intArr
    }

    val vertex = x * y
    var adjList: Array<PriorityQueue<Node>> = Array(vertex) { PriorityQueue<Node>() }

    val dx = intArrayOf(0,0,-1,1)
    val dy = intArrayOf(1,-1,0,0)

    for ( i in 0 until y) {
        for (j in 0 until x) {
            for (k in 0 until 4) {
                val nx = j + dx[k]
                val ny = i + dy[k]

                if (nx !in 0 until x || ny !in 0 until y) continue
                if (matrix[ny][nx] == 0)
                    adjList[x * i + j].add(Node(x * ny + nx, 0))
                else
                    adjList[x * i + j].add(Node(x * ny + nx, 1))
            }
        }
    }

    var distance: IntArray = IntArray(vertex) { Int.MAX_VALUE }
    val queue: PriorityQueue<Node> = PriorityQueue()
    var check: BooleanArray = BooleanArray(vertex) { false }

    distance[0] = 0
    queue.add(Node(0, 0))

    while (queue.isNotEmpty()) {
        val now = queue.poll().index

        if (check[now]) continue
        check[now] = true

        for (next in adjList[now]) {
            if (distance[next.index] > distance[now] + next.distance) {
                distance[next.index] = distance[now] + next.distance
                queue.add(Node(next.index, distance[next.index]))
            }
        }
    }

    println(distance[vertex - 1])
}

문제 풀이2 (BFS)

import java.io.BufferedReader
import java.io.InputStreamReader
import java.util.*

val dx = intArrayOf(1,-1,0,0)
val dy = intArrayOf(0,0,1,-1)

var matrix: Array<IntArray> = arrayOf()
var route: Array<IntArray> = arrayOf()

data class Dot(val x: Int, val y: Int, val distance: Int)

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {

    val (x: Int, y: Int) = readLine()
            .split(" ")
            .map { it.toInt() }

    matrix = Array(y) { intArrayOf() }
    route = Array(y) { IntArray(x) { Int.MAX_VALUE } }

    for (i in 0 until y)
        matrix[i] = readLine()
                .map { it.toInt() - 48 }
                .toIntArray()

    bfs(x, y)

}

fun bfs(x: Int, y: Int) {

    val list: LinkedList<Dot> = LinkedList()
    list.add(Dot(0, 0, 0))
    route[0][0] = 0

    while (list.isNotEmpty()) {

        val now = list.poll()

        for (i in 0 until 4) {

            val nx = now.x + dx[i]
            val ny = now.y + dy[i]

            if (nx !in 0 until x || ny !in 0 until y) continue
            if (route[ny][nx] > now.distance + matrix[ny][nx]) {
                route[ny][nx] = now.distance + matrix[ny][nx]
                list.add(Dot(nx, ny, route[ny][nx]))
            }
        }
    }
    println(route[y - 1][x - 1])
}