Kotlin中的配对比较器

时间:2018-11-22 19:51:18

标签: generics kotlin comparator

我可以在Kotlin中编写特殊类型的比较器
class Comparator() : kotlin.Comparator<Pair<Double, Int>>
但是,如何对所有扩展Comparable <...>的可能类型使用泛型编写比较器?

2 个答案:

答案 0 :(得分:1)

您可以使用帮助功能-compareBycompareByDescendingnaturalOrderreverseOrder https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.comparisons/index.html

创建比较器

例如:

val map = mapOf<Int, String>()
// ... add values to the map
val sortedMap: SortedMap<Int, String> = map.toSortedMap(compareByDescending { it })

对于您的情况:

val comparator = compareBy<Pair<Double, Int>> { it.first }

自定义比较器:

class CustomComparator<T: Comparable<T>> : Comparator<T> {
    override fun compare(o1: T, o2: T): Int {
        return o1.compareTo(o2)
    }
}

答案 1 :(得分:1)

您可以使用以下功能将两个比较器组合为一个对比较器,该对比较器将一个对的第一个值与第一个比较器进行比较,然后将第二个值与第二个比较器进行比较。

fun <T, U> pairComparator(
    firstComparator: Comparator<T>, 
    secondComparator: Comparator<U>
): Comparator<Pair<T, U>> = 
         compareBy(firstComparator) { p: Pair<T, U> -> p.first }
          .thenBy(secondComparator) { p: Pair<T, U> -> p.second }

然后按以下方式使用它:

val source = listOf(1 to 2.3, 1 to 2.0, 0 to 3.0)
val result1 = source.sortedWith(pairComparator(naturalOrder(), naturalOrder()))
println(result1)  // [(0, 3.0), (1, 2.0), (1, 2.3)]

val result2 = source.sortedWith(pairComparator(naturalOrder(), reverseOrder()))
println(result2)  // [(0, 3.0), (1, 2.3), (1, 2.0)]

在此处查看完整代码:https://pl.kotl.in/BkMOzb8CX