Question

我尝试在scala中编写一个快速功能的函数，但我不断收到java.lang.StackOverflowError。我认为当我递归地为n / 2调用此函数时，它与第三行中使用的两个斜杠有关。有人可以解释为什么会发生这种情况

def fast_power(x:Double, n:Int):Double = {
if(n % 2 == 0 && n > 1)
        fast_power(x, n/2) * fast_power(x, n /2)
else if(n % 2 == 1 && n > 1)
        x * fast_power(x, n - 1)
else if(n == 0) 1
else  1 / fast_power(x, n)
}

Answer 1

您的代码没有终止，因为n = 1没有任何案例。此外，您的fast_power具有线性运行时。

如果你把它写成这样：

def fast_power(x:Double, n:Int):Double = {
  if(n < 0) {
    1 / fast_power(x, -n)
  } else if (n == 0) {
    1.0
  } else if (n == 1) {
    x
  } else if (n % 2 == 0) {
    val s = fast_power(x, n / 2)
    s * s
  } else {
    val s = fast_power(x, n / 2)
    x * s * s
  }
}

然后很明显，运行时是对数的，因为 <{1}}在每次递归调用中至少减半。

我对n - vs - if没有任何强烈的意见，所以我只是按升序排序所有案例。

Answer 2

首选match构造而不是多个if / else块。这将帮助您隔离您遇到的问题（错误的递归调用），并编写更易理解的递归函数。始终将终止条件放在第一位。

def fastPower(x:Double, m:Int):Double = m match {
  case 0 => 1
  case 1 => x
  case n if n%2 == 0 => fastPower(x, n/2) * fastPower(x, n/2)
  case n => x * fastPower(x, n - 1)
}

scala中的快速功能

2 个答案: