Kotlin - 在索引范围内的 IntArray 中查找最小值

Question

我正在将一些较旧的 Java 代码重构到 Kotlin。有一个函数可以返回 [a, b] 范围内 Kotlin IntArray 中元素所持有的最小值的索引。范围值默认为 0，数组大小默认为 1。

我想做一些类似...的事情

return data.minOf().indexOf()

...但仅在 data 的 a 和 b 索引之间进行迭代。

函数如下：

// data is the IntArray property that I'm looping through. 
fun absMinIndex(a: Int = 0, b: Int = (data.size - 1)) : Int {
    var minVal = data[a]
    var minIndex = 0

    for (i in (a + 1)..b) {
        val e = data[i]
        if (e < minVal) {
            minVal = e
            minIndex = i
        }
    }
    return maxIndex
}

这个 [for 循环] 通过从不访问超出范围的索引以及不生成复制的数组/子数组很好地解决了这个问题。我想知道它是否可以做得“更漂亮”。

问题

是否有更惯用的 Kotlin 方法在不会对我当前解决方案的性能产生负面影响的范围内迭代数组？

为了清晰起见，编辑了一些代码。

Answer 1

我相信这种方法会更惯用：

1. 使用IntRange作为输入参数
2. 为IntArray 定义扩展方法，提供自定义迭代器以遍历所需范围内的列表，将值包装到IndexedValue：

fun IntArray.withIndexInRange(range: IntRange = 0..lastIndex) = Iterable {
    require(range.first >= 0 && range.last <= lastIndex)
    object : Iterator<IndexedValue<Int>> {
        private var index = range.first
        override fun hasNext() = index <= range.last
        override fun next() = IndexedValue(index, this@withIndexInRange[index++])
    }
}

1. 使用 stdlib 中的 minByOrNull 方法查找最小值或将其包装到另一个扩展方法中以方便使用：

fun <T : Comparable<T>> Iterable<IndexedValue<T>>.indexOfMinOrNull() = minByOrNull { it.value }?.index

用法：

data.withIndexInRange(a..b).indexOfMinOrNull()

请注意，这会产生一些性能损失（创建和 GC 的 N 个额外对象），但正如 Donald Knuth 所说：

过早的优化是万恶之源

所以，我相信更好的可读性值得。

Answer 2

正如 Tenfour04 所建议的，在不损失性能的情况下，您无能为力。但是根据你的想法改变你的调用方式，你可以把它变成一个扩展函数。

fun IntArray.findIndexOfMinInRange(a: Int = 0, b: Int = this.size - 1): Int {
    var maxVal = get(a)
    var maxIndex = 0

    for (i in (a + 1)..b) {
        if (get(i) < maxVal) {
            maxVal = get(i)
            maxIndex = i
        }
    }
    return maxIndex
}

//and call it like this: 

data.findIndexOfMinInRange(0, 15) //or without anything in the parentheses for the default values

我要更改的一件事是函数内部的变量名称，我们正在搜索最小值，而不是最大值，以及最小值的索引，而不是最大索引。也可能（可能很大）创建一个 data[it] 的 val 而不是访问它两次可能会更好（老实说不确定，我们会用一个 .get 来换取几个字节的内存）。

总而言之，我可能会留在这里：

fun IntArray.findIndexOfMinInRange(fromIndex: Int = 0, toIndex: Int = this.size - 1): Int {
    var min = get(fromIndex)
    var indexOfMin = fromIndex

    for(i in (fromIndex + 1)..toIndex){
        val current = get(i)
        if (current < min) {
            min = current
            indexOfMin = i
        }
    }

    return indexOfMin
}

//would be called in the same way the one above

另外，请注意，如果您创建了一个特定大小的IntArray，并且没有完全填充它，那么它将为未填充的默认值保留为0。如果你这样做：

val data = IntArray(6)
data[0] = 10
data[1] = 11
data[2] = 100
data[3] = 9
data[4] = 50

那么实际数组是[10, 11, 100, 9, 50, 0]。