結果

問題 No.1780 [Cherry Anniversary] 真冬に咲く26の櫻の木
ユーザー yudedako
提出日時 2022-01-27 21:08:08
言語 Scala(Beta)
(3.6.2)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,924 bytes
コンパイル時間 16,664 ms
コンパイル使用メモリ 268,208 KB
実行使用メモリ 98,532 KB
最終ジャッジ日時 2024-11-08 12:21:21
合計ジャッジ時間 88,470 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 37 TLE * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

import scala.io.StdIn.readLine
import scala.reflect.ClassTag
import scala.math.*
type Matrix = Array[Array[Option[Long]]]

extension (left: Matrix)
  inline def *(right: Matrix): Matrix =
    val row = left.length
    val column = right(0).length
    val mid = right.length
    Array.tabulate(row){i =>
      Array.tabulate(column){j =>
        (0 until mid).foldLeft(None: Option[Long]){(acc, k) =>
          (acc, left(i)(k).flatMap(l => right(k)(j).map(r => l + r))) match
            case (Some(a), Some(b)) => Some(max(a, b))
            case (None, r) => r
            case (l, None) => l
        }
      }
    }
  inline def pow(exp: Int): Matrix =
    val size = left.length
    var result = Array.tabulate(size){i => Array.tabulate(size){j => if i == j then Some(0L) else None}}
    var base = left
    var e = exp
    while e > 0 do
      if (e & 1) == 1 then
        result *= base
      base *= base
      e >>= 1
    result

@main def main =
  val initialColor = readLine().split(' ').map(_.toInt - 1)
  val maxCount = readLine().split(' ').map(_.toInt)
  val n = readLine().toInt
  val skill = Array.fill(n){
    val Array(s, a, b, e) = readLine().split(' ')
    (s, a.toInt - 1, b.toInt - 1, e.toLong)
  }
  val graph = Array.fill(26){Array.tabulate(16){i => Array.tabulate(16){j => if i == j then Some(0L) else None}}}
  for (s, a, b, e) <- skill do
    for i <- s.map(_ - 'A') do
      graph(i)(a)(b) = Some(max(graph(i)(a)(b).getOrElse(e), e))
      graph(i)(b)(a) = Some(max(graph(i)(b)(a).getOrElse(e), e))
  for i <- 0 until 26 do
    graph(i) = graph(i).pow(maxCount(i))
  var result = Long.MinValue
  for dest <- 0 until 16 do
    (0 until 26).foldLeft(Some(0L): Option[Long]){(acc, i) => acc.flatMap{a => graph(i)(initialColor(i))(dest).map(a + _)}} match
      case Some(sum) =>
        result = max(result, sum)
      case None =>
  println(if result == Long.MinValue then "Impossible" else result)
0