Loading

kotlin列表排序

一、排序的简单用法

data class User(val name: String, val isOnline: Boolean)

fun main() {
    val user1 = User("1", false)
    val user2 = User("2", false)
    val user3 = User("3", true)
    val user4 = User("4", true)
    val userList = mutableListOf(user4, user3, user2, user1)
    userList.forEach(::println)
}

输出结果如下:

User(name=4, isOnline=true)
User(name=3, isOnline=true)
User(name=2, isOnline=false)
User(name=1, isOnline=false)

我们希望按用户名进行排序,实现代码如下:

userList.sortBy { user -> user.name } // 指定以name属性进行升序排序
userList.sortByDescending { user -> user.name } // 指定以name属性进行降序排序

升序排序的打印结果如下:

User(name=1, isOnline=false)
User(name=2, isOnline=false)
User(name=3, isOnline=true)
User(name=4, isOnline=true)

Kotlin中的排序就是这么简单,如果我们把列表声明为不可变的列表,则不能使用sortBy或sortByDescending方法,因为列表不可以改变。
对于不可变列表的排序可以使用sortedBy或sortedBytDescending方法,方法名差不多,只是多了ed两个字母,它的实现是创建一个新的列表来保存排序后的结果,示例如下:

val userList = listOf(user4, user3, user2, user1)
val newList = userList.sortedBy { user -> user.name }
newList.forEach(::println)

二、排序的高级用法

查看sortBy和sortByDescending的源码发现,其是通过调用sortWith方法实现的,对应的sortedBy和sortedByDescending则是通过调用sortedWith方法实现的,sortWith的功能和sortedWith的功能原理是一样的,所以我这里讲清楚sortWith的使用即可。

使用sortWith实现升序排序

val userList = mutableListOf(user4, user3, user2, user1)
userList.sortWith(kotlin.Comparator {u1, u2 ->
    u1.name.compareTo(u2.name)
})
userList.forEach(::println)

使用sortWith实现降序排序

降序非常简单,把u1和u2的比较顺序交换一下即可

userList.sortWith(kotlin.Comparator {u1, u2 ->
    u2.name.compareTo(u1.name)
})

使用sortWith实现多重排序

sortBy只能指定一个属性参与排序,并且此属性对象必须已经实现了Comparable接口,通过调用该接口的compare方法进行排序,而使用sortWith则可以指定多个属性参与排序,属性不需要实现Comparable接口,可以自己定制排序的逻辑。

假设我们希望按用户的在线状态排序,在线的排前面,离线的排后面,状态相同的情况下再按名称排序,实现代码如下:

fun main() {
    val user1 = User("1", false)
    val user2 = User("2", false)
    val user3 = User("3", true)
    val user4 = User("4", true)
    val userList = mutableListOf(user2, user1, user4, user3)

    println("排序前:")
    userList.forEach(::println)

    userList.sortWith(kotlin.Comparator { u1, u2 ->
        if (u1.isOnline != u2.isOnline) {
            u2.isOnline.compareTo(u1.isOnline) // 状态以降序排序
        } else {
            u1.name.compareTo(u2.name)         // 名字以升序排序
        }
    })

    println("排序后:")
    userList.forEach(::println)
}

输出结果如下:
排序前:

User(name=2, isOnline=false)
User(name=1, isOnline=false)
User(name=4, isOnline=true)
User(name=3, isOnline=true)

排序后:

User(name=3, isOnline=true)
User(name=4, isOnline=true)
User(name=1, isOnline=false)
User(name=2, isOnline=false)

再一次感受到了Kotlin的强大,真是太方便了,使用Kotlin进行开发可以节省我们大量的宝贵时间。

posted @ 2022-09-05 17:39  冲锋的麦克  阅读(452)  评论(0编辑  收藏  举报