Files
kotlin-fork/compiler/testData/codegen/box/defaultArguments/kt36853_fibonacci.kt
T
Vladimir Sukharev 924898afb7 [K/N] KFC-446: K2 platform: Native alpha
Merge-request: KT-MR-7905
Merged-by: Vladimir Sukharev <Vladimir.Sukharev@jetbrains.com>
2023-01-04 16:10:40 +00:00

49 lines
1.1 KiB
Kotlin
Vendored

// WITH_STDLIB
// IGNORE_BACKEND_K2: JVM_IR, JS_IR, NATIVE
// FIR status:
// java.lang.StackOverflowError
// at Nat$Companion$invoke$1.next(kt36853_fibonacci.kt:40) ...
fun box(): String {
Nat<Int>(
nil = 0,
next = { this + 1 }
).run {
val fib = fibonacci(10)
if (fib != 89)
return "Failed: $fib"
}
return "OK"
}
fun <T> Nat<T>.fibonacci(
n: T,
seed: Pair<T, T> = nil to one,
fib: (Pair<T, T>) -> Pair<T, T> = { (a, b) -> b to a + b },
i: T = nil,
): T =
if (i == n) fib(seed).first
else fibonacci(n = n, seed = fib(seed), i = i.next())
tailrec fun <T> Nat<T>.plus(l: T, r: T, acc: T = l, i: T = nil): T =
if (i == r) acc else plus(l, r, acc.next(), i.next())
interface Nat<T> {
val nil: T
val one: T get() = nil.next()
fun T.next(): T
operator fun T.plus(t: T) = plus(this, t)
companion object {
operator fun <T> invoke(nil: T, next: T.() -> T): Nat<T> =
object: Nat<T> {
override val nil: T = nil
override fun T.next(): T = next()
}
}
}