結果

問題 No.1750 ラムドスウイルスの感染拡大-hard
ユーザー yudedakoyudedako
提出日時 2022-01-26 23:06:21
言語 Kotlin
(1.9.23)
結果
AC  
実行時間 478 ms / 2,000 ms
コード長 1,685 bytes
コンパイル時間 13,520 ms
コンパイル使用メモリ 460,760 KB
実行使用メモリ 60,456 KB
最終ジャッジ日時 2024-06-06 06:37:46
合計ジャッジ時間 28,134 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 274 ms
51,980 KB
testcase_01 AC 275 ms
51,672 KB
testcase_02 AC 271 ms
51,692 KB
testcase_03 AC 277 ms
51,600 KB
testcase_04 AC 367 ms
53,648 KB
testcase_05 AC 271 ms
51,592 KB
testcase_06 AC 277 ms
51,664 KB
testcase_07 AC 271 ms
51,876 KB
testcase_08 AC 356 ms
54,104 KB
testcase_09 AC 339 ms
53,876 KB
testcase_10 AC 348 ms
54,000 KB
testcase_11 AC 364 ms
54,072 KB
testcase_12 AC 368 ms
54,672 KB
testcase_13 AC 364 ms
54,432 KB
testcase_14 AC 470 ms
60,296 KB
testcase_15 AC 478 ms
60,456 KB
testcase_16 AC 475 ms
60,320 KB
testcase_17 AC 458 ms
59,796 KB
testcase_18 AC 463 ms
60,124 KB
testcase_19 AC 457 ms
59,820 KB
testcase_20 AC 410 ms
58,124 KB
testcase_21 AC 432 ms
59,136 KB
testcase_22 AC 346 ms
55,140 KB
testcase_23 AC 458 ms
60,264 KB
testcase_24 AC 340 ms
54,076 KB
testcase_25 AC 344 ms
55,536 KB
testcase_26 AC 323 ms
53,888 KB
testcase_27 AC 294 ms
52,656 KB
testcase_28 AC 307 ms
52,760 KB
testcase_29 AC 281 ms
51,988 KB
testcase_30 AC 378 ms
55,188 KB
testcase_31 AC 376 ms
54,860 KB
testcase_32 AC 371 ms
54,804 KB
testcase_33 AC 367 ms
54,800 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

const val MOD = 998244353

class Matrix(val row: Int, val column: Int, val state: Array<LongArray>) {
    constructor(row: Int, column: Int): this(row, column, Array(row){ LongArray(column) })
    operator fun times(other: Matrix): Matrix {
        require(column == other.row)
        val result = Array(row){LongArray(other.column)}
        for (i in 0 until row) {
            for (j in 0 until other.column) {
                var sum = 0L
                for (k in 0 until column) {
                    sum += state[i][k] * other.state[k][j] % MOD
                }
                result[i][j] = sum % MOD
            }
        }
        return Matrix(row, other.column, result)
    }
    fun pow(exp: Long): Matrix {
        require(row == column)
        var e = exp
        var result = E(row)
        var b = this
        while (e > 0) {
            if (e and 1 == 1L) {
                result *= b
            }
            e = e shr 1
            b *= b
        }
        return result
    }
    companion object {
        fun E(size: Int): Matrix {
            return Matrix(size, size, Array(size){i -> LongArray(size){j -> if (i == j) 1 else 0} })
        }
    }
}

fun main() {
    val (n, m, t) = readLine()!!.trim().split(' ').let{(n, m, t) -> Triple(n.toInt(), m.toInt(), t.toLong())}
    val roads = List(m){
        val (u, v) = readLine()!!.trim().split(' ').map(String::toInt)
        u to v
    }
    val matrix = Matrix(n, n)
    for ((u, v) in roads) {
        matrix.state[u][v] += 1L
        matrix.state[v][u] += 1L
    }
    val initial = Matrix(n, 1).also { it.state[0][0] = 1L }
    val result = matrix.pow(t) * initial
    println(result.state[0][0])
}
0