很难理解返回两个数字索引的 Scala 代码

Question

我很难理解下面代码中的 m.get 和 m+(x._1 -> x._2) ，谁能告诉我是什么会的

object Solution {
    def twoSum(nums: Array[Int], target: Int): Array[Int] = {
      nums.zipWithIndex.foldLeft(Map.empty[Int,Int])((m,x)=>
      {
          if(m.get(target - x._1)==None)
              m+(x._1 -> x._2)
        else
            return Array(m.getOrElse(target-x._1, -1), x._2)
    })
    null
  }
}

此代码返回两个数字的索引，以便它们相加为特定目标。

Answer 1

这里有几种不同的（更好的？）方法可以获得相同的结果（基本上相同）。

如果您想要其值总和为目标的所有索引对。

def twoSum(nums :Array[Int], target :Int) :Iterator[Seq[Int]] =
  nums.indices
      .combinations(2)
      .filter{case Seq(a,b) => nums(a) + nums(b) == target}

twoSum(Array(3,5,11,2,13,9), 41)  //returns empty Iterator
twoSum(Array(3,5,11,2,13,9), 14)  //returns Iterator(Seq(0, 2), Seq(1, 5))

如果您只想要总和为目标的第一对（提前终止）。

def twoSum(nums :Array[Int], target :Int) :Option[Seq[Int]] =
  nums.indices
      .combinations(2)
      .find{case Seq(a,b) => nums(a) + nums(b) == target}

twoSum(Array(3,5,11,2,13,9), 41)  //returns None
twoSum(Array(3,5,11,2,13,9), 14)  //returns Some(Seq(0, 2))

如果您想避免 Option，并且如果没有 2 个值与目标相加，则只返回一个空集合。

def twoSum(nums :Array[Int], target :Int) :Seq[Int] =
  nums.indices
      .combinations(2)
      .find{case Seq(a,b) => nums(a) + nums(b) == target}
      .getOrElse(Seq())

twoSum(Array(3,5,11,2,13,9), 41)  //returns Seq()
twoSum(Array(3,5,11,2,13,9), 14)  //returns Seq(0, 2)

Answer 2

这是解决该问题的更有效和惯用的方法。

def twoSum(nums: ArraySeq[Int], target: Int): Option[(Int, Int)] = {
  val allIndexes = for {
    i <- Iterator.range(start = 0, end = nums.length)
    j <- Iterator.range(start = i + 1, end = nums.length)
  } yield i -> j

  allIndexes.find {
    case (i, j) => (nums(i) + nums(j)) == target
  }
}

(注意： ArraySeq 与任何普通数组一样，但它是不可变的，它是在 2.13 中引入的，如果您在旧版本中，只需使用常规 @ 987654324@).

Answer 3

这是我自己的看法，该函数循环遍历所有数字对，直到找到答案。它利用了 scala 的优化递归能力。此外，它仅使用对初学者有用的简单概念：函数、数组、整数和 if-else 语句。

def twoSum(nums :Array[Int], target : Int): Array[Int] = {
    def iter(a:Int, b: Int): Array[Int] = {
        if(nums(a) + nums(b) == target) Array(a,b) 
        else if(b < nums.length-1) iter(a,b+1)  
        else iter(a+1,a+2)
    }
    iter(0,1) 
}  

jwvh：是的。失败时为空数组：

def twoSum(nums: Array[Int], target: Int): Array[Int] = {
    def iter(a: Int,b: Int): Array[Int] = {
        if(nums(a) + nums(b) == target) Array(a,b) 
        else if(b < nums.length-1) iter(a,b+1)  
        else if(a < nums.length-2) iter(a+1,a+2)
        else Array()
    }
    iter(0,1)
}