Kotlin 中的比较器

在 Kotlin 中，Comparator 接口用于定义一个比较两个对象的方法。它允许在排序和集合操作中使用自定义比较逻辑。

使用 Comparator 接口

Comparator 接口包含一个方法 compare(a: T, b: T): Int，其中 a 和 b 是要比较的对象，T 是它们的类型，Int 是比较结果。比较结果应该符合以下规则：

• 如果 a 小于 b，则返回负整数；
• 如果 a 等于 b，则返回 0；
• 如果 a 大于 b，则返回正整数。

以下示例演示如何实现一个比较器，该比较器根据字符串长度对它们进行排序：

val names = listOf("Tom", "Jerry", "Spike", "Tyke")
val nameLengthComparator = Comparator { a: String, b: String -> a.length - b.length }
val sortedNames = names.sortedWith(nameLengthComparator)
println(sortedNames) // [Tom, Tyke, Jerry, Spike]

在此示例中，我们使用 sortedWith() 函数和自定义比较器对字符串列表进行排序。该比较器定义为一个 lambda，它接受两个字符串 a 和 b，然后对它们进行比较，并返回它们长度之间的差。

简化比较器的实现

在 Kotlin 中，可以使用 compareBy() 和 thenBy() 函数轻松地创建比较器。compareBy() 函数允许您指定要比较的属性并按其顺序创建比较器。thenBy() 函数允许在比较属性相同时定义其他比较逻辑。

以下示例演示如何使用 compareBy() 和 thenBy() 函数创建比较器，该比较器按年龄和姓名对 Person 对象进行排序：

data class Person(val name: String, val age: Int)

val people =
    listOf(
        Person("Tom", 23),
        Person("Jerry", 25),
        Person("Spike", 30),
        Person("Tyke", 3)
    )

val sortedPeople =
    people.sortedWith(compareBy(Person::age).thenBy(Person::name))

println(sortedPeople)

该示例将 Person 对象列表按年龄升序排序，然后按姓名升序排序。请注意，我们使用比较器属性引用操作符 (::) 来访问对象的属性。

使用 Comparator.naturalOrder() 和 Comparator.reverseOrder()

Kotlin 还提供了方便的方法来获得自然排序和相反排序比较器。Comparator.naturalOrder() 返回用于自然排序的比较器，而 Comparator.reverseOrder() 返回按相反顺序排序的比较器。

以下示例演示如何使用这些方法：

val numbers = listOf(3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5)
val sortedNumbers = numbers.sorted()
val reversedNumbers = numbers.sortedWith(Comparator.reverseOrder())

println(sortedNumbers) // [1, 1, 2, 3, 3, 4, 5, 5, 5, 6, 9]
println(reversedNumbers) // [9, 6, 5, 5, 5, 4, 3, 3, 2, 1, 1]

总结

Kotlin 中的 Comparator 接口使您能够自定义比较逻辑以在排序和集合操作中使用。您可以使用 lambda 表达式或使用 compareBy() 和 thenBy() 函数轻松地创建比较器。此外，使用 Comparator.naturalOrder() 和 Comparator.reverseOrder() 方法可轻松获取自然排序和相反排序比较器。