結果

問題 No.1638 Robot Maze
ユーザー face4face4
提出日時 2021-08-06 22:37:25
言語 Kotlin
(2.1.0)
結果
AC  
実行時間 431 ms / 2,000 ms
コード長 1,266 bytes
コンパイル時間 17,837 ms
コンパイル使用メモリ 446,016 KB
実行使用メモリ 60,800 KB
最終ジャッジ日時 2024-09-17 02:48:46
合計ジャッジ時間 38,190 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.util.*

data class Data(val dist: Long, val pos: Int) : Comparable<Data> {
    override fun compareTo(other: Data): Int {
        return if (dist - other.dist > 0) 1 else -1
    }
}

fun main() {
    val (h, w) = readLine()!!.split(" ").map { it.toInt() }
    val line = readLine()!!.split(" ").map { it.toLong() }
    val (si, sj, ti, tj) = readLine()!!.split(" ").map { it.toInt() - 1 }
    val di = listOf(-1, 1, 0, 0)
    val dj = listOf(0, 0, 1, -1)
    val mat = (0 until h).map { readLine()!! }

    val d = List(h) { MutableList(w) { 1L shl 60 } }
    d[si][sj] = 0
    val pq = PriorityQueue<Data>()
    pq.add(Data(d[si][sj], si * w + sj))
    while (pq.isNotEmpty()) {
        val (dist, pos) = pq.poll()
        val i = pos / w
        val j = pos % w
        if (d[i][j] != dist) continue
        for (k in 0 until 4) {
            val ni = i + di[k]
            val nj = j + dj[k]
            if (ni !in 0 until h || nj !in 0 until w || mat[ni][nj] == '#') continue
            val ndist = dist + (if (mat[ni][nj] == '@') line[5] else 0) + line[k]
            if (d[ni][nj] <= ndist) continue
            d[ni][nj] = ndist
            pq.add(Data(ndist, ni * w + nj))
        }
    }
    println(if (d[ti][tj] <= line[4]) "Yes" else "No")
}
0