結果

問題 No.1558 Derby Live
コンテスト
ユーザー yakamoto
提出日時 2021-06-26 19:29:28
言語 Kotlin
(2.1.0)
結果
AC  
実行時間 922 ms / 2,000 ms
コード長 5,065 bytes
コンパイル時間 18,139 ms
コンパイル使用メモリ 466,712 KB
実行使用メモリ 97,028 KB
最終ジャッジ日時 2024-06-25 09:18:42
合計ジャッジ時間 49,265 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 33
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.kt:54:11: warning: expected performance impact from inlining is insignificant. Inlining works best for functions with parameters of functional types
  private inline fun fx(a: IntArray, b: IntArray) = run {
          ^
Main.kt:63:13: warning: expected performance impact from inlining is insignificant. Inlining works best for functions with parameters of functional types
    private inline fun ap(x: IntArray, m: IntArray) = m // 入れ替え
            ^
Main.kt:63:27: warning: parameter 'x' is never used
    private inline fun ap(x: IntArray, m: IntArray) = m // 入れ替え
                          ^
Main.kt:83:31: warning: parameter 'l' is never used
    fun query(a: Int, b: Int, l: Int = 0, r: Int = N, k: Int = 1): IntArray {
                              ^
Main.kt:83:43: warning: parameter 'r' is never used
    fun query(a: Int, b: Int, l: Int = 0, r: Int = N, k: Int = 1): IntArray {
                                          ^
Main.kt:83:55: warning: parameter 'k' is never used
    fun query(a: Int, b: Int, l: Int = 0, r: Int = N, k: Int = 1): IntArray {
                                                      ^
Main.kt:194:11: warning: expected performance impact from inlining is insignificant. Inlining works best for functions with parameters of functional types
  private inline fun debug(a: LongArray) {
          ^
Main.kt:198:11: warning: expected performance impact from inlining is insignificant. Inlining works best for functions with parameters of functional types
  private inline fun debug(a: IntArray) {
          ^
Main.kt:202:11: warning: expected performance impact from inlining is insignificant. Inlining works best for functions with parameters of functional types
  private inline fun debug(a: BooleanArray) {
          ^
Main.kt:206:11: warning: expected performance impact from inlining is insignificant. Inlining works best for functions with parameters of functional types
  private inline fun toString(a: BooleanArray) = run{a.map { if (it) 1 else

ソースコード

diff # 

import java.io.BufferedReader
import java.io.InputStream
import java.io.InputStreamReader
import java.lang.AssertionError
import java.util.*
import kotlin.math.abs
import kotlin.math.max
import kotlin.math.min

val MOD = 1_000_000_007

class Solver(stream: InputStream, private val out: java.io.PrintWriter) {
  private val reader = BufferedReader(InputStreamReader(stream), 32768)

  private val NN = ni()
  private val M = ni()
  private val Q = ni()
  fun solve() {
    val zero = IntArray(NN) { it }
    val tree = SegmentTreeA(M, zero)

    for (q in 0 until Q) {
      val type = ni()
      when (type) {
        1 -> {
          val d = ni() - 1
          val p = na(NN, -1)
          tree.update(d, p)
        }
        2 -> {
          val s = ni()
          val p = tree.query(0, s)
          debug { "type2 ${p.joinToString(" ")}" }
          val inv = IntArray(NN)
          for (i in 0 until NN) {
            inv[p[i]] = i
          }
          out.println(inv.map { it + 1 }.joinToString(" "))
        }
        else -> {
          val (l, r) = na(2, -1)
          val p = tree.query(l, r + 1)
          debug { "type3 ${p.joinToString(" ")}" }
          var ans = 0L
          for (i in 0 until NN) {
            ans += abs(i - p[i])
          }
          out.println(ans)
        }
      }
    }
  }

  private inline fun fx(a: IntArray, b: IntArray) = run {
    val res = IntArray(NN)
    for (i in res.indices) {
      res[i] = b[a[i]]
    }
    res
  }

  inner class SegmentTreeA(val n: Int, val zero: IntArray) {
    private inline fun ap(x: IntArray, m: IntArray) = m // 入れ替え

    private val N =
      if (Integer.highestOneBit(n) == n) n
      else Integer.highestOneBit(n) shl 1

    private val dat = Array(2*N){zero}

    fun update(i: Int, a: IntArray) {
      var ix = i + N
      dat[ix] = ap(dat[ix], a)
      while(ix > 1) {
        dat[ix shr 1] = if (ix and 1 == 0) fx(dat[ix], dat[ix xor 1]) else fx(dat[ix xor 1], dat[ix])
        ix = ix shr 1
      }
    }

    /**
     * [a, b)
     */
    fun query(a: Int, b: Int, l: Int = 0, r: Int = N, k: Int = 1): IntArray {
      var resL = zero
      var resR = zero
      var left = a + N
      var right = b - 1 + N

      while(left <= right) {
        if ((left and 1) == 1) resL = fx(resL, dat[left])
        if ((right and 1) == 0) resR = fx(dat[right], resR)
        left = (left + 1) shr 1 // 右の子供なら右の親に移動
        right = (right - 1) shr 1 // 左の子供なら左の親に移動
      }

      return fx(resL, resR)
    }

    fun get(i: Int) = dat[N + i]

    override fun toString() = "[${(0 until n).map{get(it)}.joinToString(" ")}]"
  }













































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

  private var tokenizer: StringTokenizer? = null
  private fun next(): String {
    while (tokenizer == null || !tokenizer!!.hasMoreTokens()) {
      tokenizer = StringTokenizer(reader.readLine())
    }
    return tokenizer!!.nextToken()
  }

  private fun ni() = next().toInt()
  private fun nl() = next().toLong()
  private fun ns() = next()
  private fun na(n: Int, offset: Int = 0): IntArray {
    return IntArray(n) { ni() + offset }
  }
  private fun nal(n: Int, offset: Int = 0): LongArray {
    val res = LongArray(n)
    for (i in 0 until n) {
      res[i] = nl() + offset
    }
    return res
  }

  private fun na2(n: Int, offset: Int = 0): Array<IntArray> {
    val a  = Array(2){IntArray(n)}
    for (i in 0 until n) {
      for (e in a) {
        e[i] = ni() + offset
      }
    }
    return a
  }

  private inline fun debug(msg: () -> String) {
    if (isDebug) System.err.println(msg())
  }

  /**
   * コーナーケースでエラー出たりするので、debug(dp[1])のように添え字付きの場合はdebug{}をつかうこと
   */
  private inline fun debug(a: LongArray) {
    debug { a.joinToString(" ") }
  }

  private inline fun debug(a: IntArray) {
    debug { a.joinToString(" ") }
  }

  private inline fun debug(a: BooleanArray) {
    debug { toString(a) }
  }

  private inline fun toString(a: BooleanArray) = run{a.map { if (it) 1 else 0 }.joinToString("")}

  private inline fun debugDim(A: Array<LongArray>) {
    if (isDebug) {
      for (a in A) {
        debug(a)
      }
    }
  }
  private inline fun debugDim(A: Array<IntArray>) {
    if (isDebug) {
      for (a in A) {
        debug(a)
      }
    }
  }
  private inline fun debugDim(A: Array<BooleanArray>) {
    if (isDebug) {
      for (a in A) {
        debug(a)
      }
    }
  }

  /**
   * 勝手にimport消されるのを防ぎたい
   */
  private fun hoge() {
    min(1, 2)
    max(1, 2)
    abs(-10)
  }

  private inline fun assert(b: Boolean) = run{if (!b) throw AssertionError()}
  private inline fun assert(b: Boolean, f: () -> String) = run{if (!b) throw AssertionError(f())}
}

fun main() {
  val out = java.io.PrintWriter(System.out)
  Solver(System.`in`, out).solve()
  out.flush()
}
0