Scala:列表中的折叠问题的问题



我正在编写一个scala函数,该函数返回列表中均匀元素的总和,列表中奇数元素的减总和。我不能使用突变物,递归或/while循环作为解决方案。下面的代码通过2/3测试,但我似乎无法弄清楚为什么它不能正确计算最后一个测试。

def sumOfEvenMinusOdd(l: List[Int]) : Int = {
    if (l.length == 0) return 0
    val evens = l.filter(_%2==0)
    val odds = l.filter(_%2==1)
    val evenSum = evens.foldLeft(0)(_+_)
    val oddSum = odds.foldLeft(0)(_+_)
    evenSum-oddSum
  }
//BEGIN TESTS
val i1 = sumOfEvenMinusOdd(List(1,3,5,4,5,2,1,0)) //answer: -9
val i2 = sumOfEvenMinusOdd(List(2,4,5,6,7,8,10)) //answer: 18
val i3 = sumOfEvenMinusOdd(List(109, 19, 12, 1, -5, -120, -15, 30,-33,-13, 12, 19, 3, 18, 1, -1)) //answer -133

我的代码正在输出以下内容:

defined function sumOfEvenMinusOdd
i1: Int = -9
i2: Int = 18
i3: Int = -200

我感到非常困惑,为什么这些负数绊倒了我的其余代码。我看到了一篇文章,用foldleft foldright解释了操作顺序,但即使更改为foldright仍然产生i3:int = -200。我缺少一个细节吗?任何指导/帮助都将不胜感激。

问题不是foldLeftfoldRight,问题是您过滤奇数的方式:

val odds = l.filter(_ % 2 == 1)

应该是:

val odds = l.filter(_ % 2 != 0)

谓词_ % 2 == 1仅对积极元素产生True。例如,表达式-15 % 2等于-1,而不是1

作为旁注,我们还可以使其更有效:

def sumOfEvenMinusOdd(l: List[Int]): Int = {
  val (evenSum, oddSum) = l.foldLeft((0, 0)) {
    case ((even, odd), element) =>
      if (element % 2 == 0) (even + element, odd) else (even, odd + element)
  }
  evenSum - oddSum
}

或什至更好地累积差异:

def sumOfEvenMinusOdd(l: List[Int]): Int = {
  l.foldLeft(0) {
    case (diff, element) =>
      diff + element * (if (element % 2 == 0) 1 else -1)
  }
}

问题在您在列表上应用以查找奇数的过滤条件上。由于mod 2返回-1,因此您不适用于负数奇数。

number % 2 == 0 if number is even
number % 2 != 0 if number is odd

因此,如果您更改过滤器条件,则全部按预期工作。

另一个建议:当您可以直接使用总和函数时,为什么要使用foldleft函数进行简单的总和操作?

  test("Test sum Of even minus odd") {
    def sumOfEvenMinusOdd(l: List[Int]) : Int = {
      val evensSum = l.filter(_%2 == 0).sum
      val oddsSum = l.filter(_%2 != 0).sum
      evensSum-oddsSum
    }
    assert(sumOfEvenMinusOdd(List.empty[Int]) == 0)
    assert(sumOfEvenMinusOdd(List(1,3,5,4,5,2,1,0)) == -9) //answer: -9
    assert(sumOfEvenMinusOdd(List(2,4,5,6,7,8,10)) == 18) //answer: 18
    assert(sumOfEvenMinusOdd(List(109, 19, 12, 1, -5, -120, -15, 30,-33,-13, 12, 19, 3, 18, 1, -1)) == -133)
  }

使用此解决方案,您的功能更加清晰,您可以在Funciton上删除IF

最新更新