迹忆客 专注技术分享

Kotlin 库提供了开发人员可以根据需求利用的不同方式。 本教程将教我们如何使用单个字段和多个字段对集合进行排序。

生成 Kotlin 项目

打开 IntelliJ 开发环境并选择 File > New > Project。 输入项目名称 kotlin-sortby，在语言部分选择 Kotlin，在构建系统部分选择 Intellij。

最后，按创建按钮生成项目。 在 src/main/kotlin 文件夹下创建一个名为 Main.kt 的文件，并将以下代码复制并粘贴到该文件中。

代码：

import java.time.Year

data class Book(val id: Int,
                val name: String,
                val yearPublished: Year)

fun fetchBooks(): ArrayList<Book>{
    return arrayListOf(
        Book(1,"Java Arrays", Year.of(1984)),
        Book(2,"Python Arrays",Year.of(2021)),
        Book(3,"Pascal for loop",Year.of(2022)),
        Book(4,"Pascal Arrays",Year.of(2021)),
        Book(5,"Java for loop",Year.of(2022))
    )
}

在这段代码中，我们创建了一个名为 fetchBooks() 的方法，它返回使用 Book 数据类创建的书籍对象列表。 我们在示例中使用此列表来了解如何根据不同的标准对书籍进行排序。

使用单个属性对集合进行排序

将以下示例代码复制并粘贴到 Main.kt 文件中的 fetchBooks() 方法之后。

使用 sortBy() 方法对集合进行排序

在下面的示例中，我们将创建一个名为 usingSortBy() 的方法，该方法使用 sortBy() 方法对我们之前使用 yearPublished 字段创建的列表进行排序。

sortBy() 方法是 MutableList 的扩展函数，不返回任何值。 该函数在应用于它的选择器函数之前有一个 crossinline 关键字，这意味着该选择器函数不允许非本地返回。

请注意，选择器函数在幕后使用 Comparable 接口对列表中的元素进行排序。

代码：

fun usingSortBy(){
    val books = fetchBooks()
    books.sortBy(Book::yearPublished)
    books.forEach(::println);
}

fun main(){
    usingSortBy();
}

输出:

Book(id=1, name=Java Arrays, yearPublished=1984)
Book(id=2, name=Python Arrays, yearPublished=2021)
Book(id=4, name=Pascal Arrays, yearPublished=2021)
Book(id=3, name=Pascal for loop, yearPublished=2022)
Book(id=5, name=Java for loop, yearPublished=2022)

使用 sortedBy() 方法对集合进行排序

集合框架提供了一个替代 sortBy() 方法的方法，它允许我们返回一个我们在这个例子中无法获得的新列表。 我们可以使用 sortedBy() 方法，它是 Iterable 接口的扩展函数。

代码：

fun usingSortedBy(){
    fetchBooks().sortedBy(Book::yearPublished)
        .forEach(::println);
}

fun main(){
    usingSortedBy();
}

输出:

Book(id=1, name=Java Arrays, yearPublished=1984)
Book(id=2, name=Python Arrays, yearPublished=2021)
Book(id=4, name=Pascal Arrays, yearPublished=2021)
Book(id=3, name=Pascal for loop, yearPublished=2022)
Book(id=5, name=Java for loop, yearPublished=2022)

sortedBy() 方法返回一个列表，其中包含使用选择器函数排序的元素，我们可以在返回的列表上调用 forEach() 方法以将对象记录到控制台。

使用多个属性对集合进行排序

在前面的示例中，我们已经了解了如何使用单个字段对元素进行排序。 但是，在某些情况下，客户端可能希望基于应用程序中的多个字段进行排序。

在本节中，我们将学习如何实现此功能。

使用 sortWith() 方法对集合进行排序

在此示例中，我们创建了一个名为 usingSortWith() 的方法，该方法使用 sortWith() 方法使用书名后跟书年对列表进行排序。

sortWith() 方法是 MutableList 的扩展函数，它接受返回比较器的任何参数。 请注意，此方法不返回任何值。

代码：

fun usingSortWith(){
    val books = fetchBooks();
    books.sortWith(compareBy(Book::name, Book::yearPublished));
    books.forEach(::println);
}

fun main(){
    usingSortWith();
}

我们使用 compareBy() 方法作为 sortWith() 方法的参数。 compareBy() 方法返回一个 Comparator 并且可以用多个选择器函数重载以根据不同的属性对列表进行排序。

在选择器函数定义之前添加的 vararg 关键字表示我们可以传递多个参数。

输出：

Book(id=1, name=Java Arrays, yearPublished=1984)
Book(id=5, name=Java for loop, yearPublished=2022)
Book(id=4, name=Pascal Arrays, yearPublished=2021)
Book(id=3, name=Pascal for loop, yearPublished=2022)
Book(id=2, name=Python Arrays, yearPublished=2021)

使用 sortedWith() 方法对集合进行排序

集合框架提供了一种替代 sortWith() 方法的方法，它允许我们返回一个包含已排序元素的新列表。 我们可以使用 sortedWith() 方法来实现这个功能。

sortedWith() 方法是 Iterable 接口的扩展函数，它接受返回 Comparator 的任何参数。

代码：

fun usingSortedWith(){
    fetchBooks().sortedWith(compareBy(Book::name, Book::yearPublished))
        .forEach(::println);
}

fun main(){
    usingSortedWith();
}

我们使用了 compareBy() 返回一个比较器，就像我们在前面的例子中一样，并重载它以使用名称和年份对列表进行排序。

输出：

Book(id=1, name=Java Arrays, yearPublished=1984)
Book(id=5, name=Java for loop, yearPublished=2022)
Book(id=4, name=Pascal Arrays, yearPublished=2021)
Book(id=3, name=Pascal for loop, yearPublished=2022)
Book(id=2, name=Python Arrays, yearPublished=2021)

总结

在本篇文章中，我们学习了根据所需标准对列表进行排序的不同方法。 我们介绍的方法包括使用 sortBy() 方法、使用 sortedBy() 方法、使用 sortWith() 方法和使用 sortedWith() 方法。

请注意 ，其他两种排序策略允许我们按降序对列表进行排序，它们包括 sortByDescending() 和 sortedByDescending()。