Scala:foldLeft的问题,列表中有负数

问题描述 投票:1回答:2

我正在编写一个Scala函数,它返回列表中偶数元素的总和,减去列表中奇数元素的总和。我不能为我的解决方案使用mutable,递归或for / 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。有没有我遗漏的细节?任何指导/帮助将不胜感激。

scala filter pattern-matching fold
2个回答
6
投票

问题不是foldLeftfoldRight,问题是你过滤掉奇数值的方式:

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

应该:

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

谓词_ % 2 == 1只会对正元素产生真实。例如,表达式-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)
  }
}

0
投票

问题在于您在列表上应用的过滤条件以查找奇数。奇怪的条件是你不适用于负奇数,因为mod 2为这种数字返回-1。

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

所以如果你改变过滤条件,一切都按预期工作。

另一个建议:当你可以直接使用sum函数时,为什么要使用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)
  }

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

© www.soinside.com 2019 - 2024. All rights reserved.