結果

問題 No.871 かえるのうた
ユーザー yakamoto
提出日時 2019-08-31 00:38:30
言語 Kotlin
(2.1.0)
結果
AC  
実行時間 879 ms / 2,000 ms
コード長 1,289 bytes
コンパイル時間 16,674 ms
コンパイル使用メモリ 463,708 KB
実行使用メモリ 93,056 KB
最終ジャッジ日時 2024-11-30 10:54:46
合計ジャッジ時間 37,901 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

import kotlin.math.max
import kotlin.math.min

// なんか本番でエラーでる
private val isDebug = runCatching {
  System.getenv("MY_DEBUG") != null
}.fold({it}, {false})

private fun readLn() = readLine()!!
private fun readInt() = readLn().toInt()
private fun readStrings() = readLn().split(" ")
private fun readInts() = readStrings().map { it.toInt() }.toIntArray()
private fun readLongs() = readStrings().map { it.toLong() }.toLongArray()
private fun debug(msg: () -> String) {
  if (isDebug) System.err.println(msg())
}
private fun debug(a: LongArray) {
  if (isDebug) debug{a.joinToString(" ")}
}

val MOD = 1000000007

data class Entry(val i: Int, val x: Long)

fun main() {
  var (N, K) = readInts()
  K--
  val X = readLongs()
  val A = readLongs()
  val set = java.util.TreeSet<Entry>(Comparator<Entry> { o1, o2 ->
    o1.x.compareTo(o2.x)
  })
  for (i in 0 until N) {
    set.add(Entry(i, X[i]))
  }

  var ans = 0
  var l = X[K]
  var r = X[K]
  while(true) {
    val e1 = set.higher(Entry(-1, l - 1))
    val e2 = set.lower(Entry(-1, r + 1))
    debug{"l:$l r:$r"}
    if (e1 != null && e2 != null && e1.x <= e2.x) {
      set.remove(e1)
      l = min(l, e1.x - A[e1.i])
      r = max(r, e1.x + A[e1.i])
      ans++
    } else {
      break
    }
  }

  println(ans)
}
0