Stanislav Erokhin
41 lines
863 B
Kotlin
41 lines
863 B
Kotlin
package foo
|
|
|
|
// CHECK_CONTAINS_NO_CALLS: maxBySquare
|
|
|
|
data class Result(var value: Int = 0, var invocationCount: Int = 0)
|
|
|
|
inline fun maxBy(a: Array<Int>, inline keyFun: (Int) -> Int): Int {
|
|
var maxVal = a[0]
|
|
var maxKey = keyFun(maxVal)
|
|
|
|
for (element in a) {
|
|
val key = keyFun(element)
|
|
|
|
if (key > maxKey) {
|
|
maxVal = element
|
|
maxKey = key
|
|
}
|
|
}
|
|
|
|
return maxVal
|
|
}
|
|
|
|
fun maxBySquare(a: Array<Int>, r: Result): Result {
|
|
var invocationCount = 0
|
|
val maxVal = maxBy(a, { x -> invocationCount++; x * x;})
|
|
|
|
r.value = maxVal
|
|
r.invocationCount = invocationCount
|
|
|
|
return r
|
|
}
|
|
|
|
fun box(): String {
|
|
var r1 = maxBySquare(array(1,2,3,4,5), Result())
|
|
assertEquals(Result(5, 6), r1)
|
|
|
|
var r2 = maxBySquare(array(-5,1,2,3,4), Result())
|
|
assertEquals(Result(-5, 6), r2)
|
|
|
|
return "OK"
|
|
} |