从 StdIn 读取输入，要求在运行时提高性能，请求建议

Question

我正在尝试在黑客地球中提交以下 scala 解决方案，记住我！。但是，由于超出了时间限制，它没有被占用。

关于如何提高运行时间的任何建议？

我看到计时器，循环中的 readInt（第二个 for 循环）需要更多时间，随着输入的增加......需要考虑我可以使用其他任何方法来改进吗？

import java.util.concurrent.TimeUnit

import scala.collection.mutable.{ArrayBuffer, Map}
import scala.io.StdIn.{readInt, readLine}

import scala.io.Source.stdin

object MemorizeMe {

  def main(args: Array[String]): Unit = {

    var inpLen = readInt()
    var inp = readLine().split(" ").map(_.toInt)

    val numQueries = readInt()
    var queryList = new ArrayBuffer[Int]

    var numsMap = Map[Int, Int]()
    time {for (i <- inp) if (numsMap.contains(i)) numsMap(i) = numsMap(i) + 1 else numsMap += (i -> 1)}

    time{for (i <- 1 to numQueries) queryList += readInt()}
    time{for(i <- queryList) if (numsMap.contains(i)) println(numsMap(i)) else println("NOT PRESENT")}

    /*for (x <- io.Source.stdin.getLines) {
      val i = x.toInt
      if (numsMap.contains(i)) println(numsMap(i)) else println("NOT PRESENT")
    }*/

  }

  def time[R](block: => R): R = {
    val t0 = System.nanoTime()
    val result = block    // call-by-name
    val t1 = System.nanoTime()
    println(TimeUnit.SECONDS.convert((t1 - t0), TimeUnit.NANOSECONDS) + "secs")
    result
  }

}

Answer 1

您有关于您将收到多少个整数的信息。您可以使用该信息创建一个固定大小的 Array，并将附加 (queryList += readInt()) 更改为更新 (queryList(i) = readInt())。

您可以在 Scala 文档 (https://docs.scala-lang.org/overviews/collections/performance-characteristics.html) 中找到附加成本为 aC，同时更新成本为 C。

Answer 2

未经测试，但这应该可以工作：

val queryList = Array.fill(numQueries)(ReadInt())

这将分配一个正确大小的Array，然后为每个元素调用ReadInt。