我有以下使用TCO的功能:
tailrec fun superDigit(n: String): Int {
val sum = n.fold(0) { sum, char -> sum + char.toString().toInt() }
return if (sum < 10) sum else superDigit(sum.toString())
}
如果我实现与这样的扩展函数相同的功能:
fun String.superDigit(): Int {
val sum = fold(0) { sum, char -> sum + char.toString().toInt() }
return if (sum < 10) sum else sum.toString().superDigit()
}
扩展函数尾调用也是优化的吗?
IMO调用扩展函数仍然是一个常规函数调用,this
作为参数,所以它仍然是一个递归调用,因为tailrec不能用于扩展函数,我的假设是它没有被编译器优化。这个假设是否正确?
因为tailrec不能用于扩展功能
你确定吗?
我刚刚通过查看IntelliJ IDEA中的Kotlin字节码来测试它。首先,扩展函数上的tailrec
代码编译成功。更进一步:比较下面的两个Kotlin代码和字节码,一个用tailrec
,另一个用没有。
科特林:
fun Double.tailrecTestExtension(): Double
= (this - 1.0).tailrecTestExtension()
字节码:
// access flags 0x19
public final static tailrecTestExtension(D)D
// annotable parameter count: 1 (visible)
// annotable parameter count: 1 (invisible)
L0
LINENUMBER 13 L0
DLOAD 0
DCONST_1
DSUB
INVOKESTATIC com/example/TestKt.tailrecTestExtension (D)D
DRETURN
L1
LOCALVARIABLE $receiver D L0 L1 0
MAXSTACK = 4
MAXLOCALS = 2
科特林:
tailrec fun Double.tailrecTestExtension(): Double
= (this - 1.0).tailrecTestExtension()
字节码:
// access flags 0x19
public final static tailrecTestExtension(D)D
// annotable parameter count: 1 (visible)
// annotable parameter count: 1 (invisible)
L0
LINENUMBER 13 L0
DLOAD 0
DCONST_1
DSUB
DSTORE 0
GOTO L0
L1
LOCALVARIABLE $receiver D L0 L1 0
MAXSTACK = 4
MAXLOCALS = 2
请注意,在第一个示例中,有INVOKESTATIC
调用(对应于常规递归),在第二个版本中用常规跳转(GOTO
)替换(对应于tailrec
引入的循环 - 预期行为)。
注意:我不是Kotlin字节码的专家,我的理解是基于汇编语言的一些基本知识。在这里,我假设这些知识可以转移到Kotlin字节码。