結果

問題 No.973 余興
ユーザー yakamoto
提出日時 2020-01-18 16:37:46
言語 Kotlin
(2.1.0)
結果
WA  
実行時間 -
コード長 2,023 bytes
コンパイル時間 14,947 ms
コンパイル使用メモリ 453,360 KB
実行使用メモリ 181,764 KB
最終ジャッジ日時 2024-06-27 16:21:21
合計ジャッジ時間 56,414 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 31 WA * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

import kotlin.math.min

val MOD = 1_000_000_007

fun main() {
  val (N, X) = readInts()
  val A = readInts()

  val dpL = Array(N){IntArray(N - it + 1)}
  val dpR = Array(N){IntArray(it + 2)}
  val mxR = run {
    val res = IntArray(N)
    var sum = 0L
    var r = N - 1
    for (l in N - 1 downTo 0) {
      sum += A[l]
      while (r > l && sum > X) {
        sum -= A[r]
        r--
      }
      res[l] = r - l + 1
    }
    res
  }
  debug(mxR)
  val mxL = run {
    val res = IntArray(N)
    var sum = 0L
    var l = 0
    for (r in 0 until N) {
      sum += A[r]
      while (l < r && sum > X) {
        sum -= A[l]
        l++
      }
      res[r] = r - l + 1
    }
    res
  }
  debug(mxL)

  for (d in 1 until N) {
    for (l in 0 until N - d) {
      val r = l + d
      val llen = min(d, mxR[l])
      val lwin = dpL[l][d] - dpL[l][d-llen] < llen
      val rlen = min(d, mxL[r])
      val rwin = dpR[r][d] - dpR[r][d-rlen] < rlen
      val win = if (lwin or rwin) 1 else 0
//      debug{"$l $r $win"}
      dpL[l][d+1] = dpL[l][d] + win
      dpR[r][d+1] = dpR[r][d] + win
    }
  }

  debugDim(dpL)
  debugDim(dpR)

  if (dpL[0][N] - dpL[0][N-1] == 1) println("A")
  else println("B")
}












































private val isDebug = try {
  // なんか本番でエラーでる
  System.getenv("MY_DEBUG") != null
} catch (t: Throwable) {
  false
}

private fun readLn() = readLine()!!
private fun readStrings() = readLn().split(" ")
private fun readInts() = readStrings().map { it.toInt() }.toIntArray()
private fun readLongs() = readStrings().map { it.toLong() }.toLongArray()
private inline fun debug(msg: () -> String) {
  if (isDebug) System.err.println(msg())
}
private fun debug(a: LongArray) {
  debug{a.joinToString(" ")}
}
private fun debug(a: IntArray) {
  debug{a.joinToString(" ")}
}
private fun debug(a: BooleanArray) {
  debug{a.map{if(it) 1 else 0}.joinToString("")}
}
private fun debugDim(A: Array<IntArray>) {
  if (isDebug) {
    for (a in A) {
      debug(a)
    }
  }
}
0