[백준] 2206번: 벽 부수고 이동하기 - kotlin

문제

https://www.acmicpc.net/problem/2206

풀이

• 벽을 뚫었는지 안뚫었는지 확인할 수 있는 변수를 설정 -> chance
• 벽을 만났을 때 벽을 아직 뚫지 않았으면 뚫고 들어간다.
• 벽이 아닌 곳을 만났을 때에는 방문했는지만 확인하고 BFS 를 진행한다.

import java.util.*

fun main() {
    val br = System.`in`.bufferedReader()
    val bw = System.out.bufferedWriter()
    // n : 가로, m : 세로
    val (n, m) = br.readLine().split(" ").map { it.toInt() }
    val graph = mutableListOf<List<Int>>()
    repeat(n) {
        graph.add(br.readLine().split("").filter { it.isNotBlank() }.map { it.toInt() })
    }
    val queue: Queue<Triple<Int, Int, Int>> = LinkedList()
    val visited = Array(n) { Array(m) { Array(2) { 0 } } }
    val dx = listOf(0, 0, 1, -1)
    val dy = listOf(1, -1, 0, 0)
    var result = -1
    queue.add(Triple(0, 0, 1))
    visited[0][0][1] = 1
    while (queue.isNotEmpty()) {
        val (y, x, chance) = queue.poll()
        if (y == n - 1 && x == m - 1) {
            result = visited[y][x][chance]
            break
        }
        for (i in 0..3) {
            val ny = y + dy[i]
            val nx = x + dx[i]
            if (nx in 0 until m && ny in 0 until n) {
                if (graph[ny][nx] == 1 && chance == 1) {
                    queue.add(Triple(ny, nx, 0))
                    visited[ny][nx][0] = visited[y][x][1] + 1
                }
                if (graph[ny][nx] == 0 && visited[ny][nx][chance] == 0) {
                    queue.add(Triple(ny, nx, chance))
                    visited[ny][nx][chance] = visited[y][x][chance] + 1
                }
            }
        }
    }
    bw.write("$result")
    br.close()
    bw.close()
}

더 좋은 방법 있으면 댓글 달아주세요!!!