我正在编写一个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。我缺少一个细节吗?任何指导/帮助都将不胜感激。
问题不是foldLeft
或foldRight
,问题是您过滤奇数的方式:
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