Scala 将 IndexedSeq[AnyVal] 转换为 Array[Int]

Question

我正在尝试使用 Scala 解决 Codility 的 GenomicRangeQuery 问题，为此我编写了以下函数：

def solution(s: String, p: Array[Int], q: Array[Int]): Array[Int] = {
  for (i <- p.indices) yield {
    val gen = s.substring(p(i), q(i) + 1)
    if (gen.contains('A')) 1
    else if (gen.contains('C')) 2
    else if (gen.contains('G')) 3
    else if (gen.contains('T')) 4
  }
}

我没有做很多测试，但似乎可以解决问题。

我的问题是 for 理解返回 scala.collection.immutable.IndexedSeq[AnyVal]，而函数必须返回 Array[Int]，因此它抛出了 type mismatch error。

有没有办法让 for 理解返回 Array[Int] 或将 IndexedSeq[AnyVal] 转换为 Array[Int]？

Answer 1

sheunis 上面的回答主要涵盖了它。

您可以通过调用toArray 将IndexedSeq 强制转换为Array，因此第一部分非常简单。对于第二部分，由于可能存在一个逻辑分支，您可以在其中遍历所有 if... else... 案例，因此您的 yield 可能会返回 Int 和 Unit 类型，它们最接近的共同祖先是 AnyVal .

请注意，如果您将 if... else... 替换为模式匹配，则会显式收到编译器警告，因为您没有捕获所有可能的 case。

gen match { case _ if gen.contains("A") => 1 case _ if gen.contains("C") => 2 ... // Throws warning unless you include a `case _ =>` with no `if` clause }

Answer 2

def solution(s: String, p: Array[Int], q: Array[Int]): Array[Int] = {
  (for (i <- p.indices) yield {
    val gen = s.substring(p(i), q(i) + 1)
    if (gen.contains('A')) 1
    else if (gen.contains('C')) 2
    else if (gen.contains('G')) 3
    else 4
  }).toArray
}

if 语句的问题是没有默认值，这就是为什么你得到的 IndexedSeq 是 Any 而不是 Int。

Answer 3

这里有两个问题，第一个来自p.indices，它返回scala.collection.immutable.Range 而不是Array。执行p.indices.toArray（或像@sheunis 建议的那样在最后添加.toArray）解决了这个问题。

另一个问题来自您的 if 语句不完整，如果所有条件都为假，您的方法将返回 (): Unit （由编译器添加）。添加默认情况，例如 else -1 作为最后一个语句应该可以解决第二个问题。

编辑：如果默认大小写永远无法追加，您可以如下抛出异常：

else {
  val err = "the input String can only contain the characters ACGT"
  throw new IllegalArgumentException(err)
}

这会通知下一个程序员和编译器你的代码中发生了什么。请注意throw 表达式的类型为Nothing，因此当计算(Int, Int, Int, Nothing) 的最小上限时，正确地产生Int，不像(Int, Int, Int, Unit)，它被润滑到AnyVal。