Kotlin中的Map排序

tech2026-02-04  1

示例代码如下:

fun main() { val map = sortedMapOf(1 to 2, 3 to 1, 2 to 3) println(map) // 输出:{1=2, 2=3, 3=1} println(map.javaClass.simpleName) // 输出:TreeMap }

从上面例子可以看到,有序Map的实现类是TreeMap,它默认是对key进行升序排序。TreeMap的构造方法中可以接收一个比较器,是用于比较key的,所以如果我们希望降序,则可以通过传入比较器实现,代码如下:

fun main() { val comparator = kotlin.Comparator { key1: Int, key2: Int -> key2.compareTo(key1) } val map = TreeMap<Int, Int>(comparator) map[1] = 2 map[3] = 1 map[2] = 3 println(map) // 输出:{3=1, 2=3, 1=2} }

OK,上面代码实现了对key的降序排序。但是有时候,我们的排序需要value的参与,比如有如下数据结构:

/** 员工类 */ data class Employee(val name: String, val isOnline: Boolean)

如上是一个员工类,有姓名属性和在线状态属性,比如一个聊天软件,登录时是在线状态,离线时就是离线状态。

我们可以用一个List来表示 一个部门的所有员工,可以用一个Map<String, List>来表示一个公司多个部门的所有员工,代码示例如下:

fun main() { val 研发部人员列表 = listOf(Employee("01研发部张某某", false), Employee("01研发部王某某", false)) val 销售部人员列表 = listOf(Employee("02销售部陈某某", true), Employee("02销售部李某某", false)) val 售后部人员列表 = listOf(Employee("03售后部周某某", false), Employee("03售后部赵某某", true)) val company = TreeMap<String, List<Employee>>() company["01研发部"] = 研发部人员列表 company["02销售部"] = 销售部人员列表 company["03售后部"] = 售后部人员列表 company.forEach { key, value -> println(key) } }

打印结果如下:

01研发部 02销售部 03售后部

这里打印的是部门的名称,可以看到默认是按部门名称进行升序排序的,此时我想按在线状态来排序部门,有在线状态员工的部门排在前面,如上3个部门中,研发部没有在线的人员,所以研发部应该是排在最后的,此时用key来做排序条件已经不能满足我们的需求了,Comparator对象只能对key进行排序,那如何实现使用value进行排序呢?答:自定义TreeMap的排序方法即可,但是查阅源码发现,compare方法是final类型的,不能覆盖,哎,看来TreeMap实现不了,因为它只能对key进行排序,除非我们把value的值也绑定到key中,但是这样的话,我们还不如直接使用List来实现,增加一个部门类,实现如下:

/** 员工类 */ data class Employee(val name: String, val isOnline: Boolean) /** 部门类 */ data class Department(val name: String, val employeeList: List<Employee>)

这里的Department类就相当于TreeMap类,name属性对应TreeMap的key,employeeList属性对应TreeMap的value,然后我们把多个Department保存到List中,这样就可以对List进行排序了,排序的时候就可以使用到员工列表对象了,实现代码如下:

fun main() { val 研发部人员列表 = mutableListOf(Employee("01研发部张某某", false), Employee("01研发部王某某", false)) val 销售部人员列表 = mutableListOf(Employee("02销售部陈某某", true), Employee("02销售部李某某", false)) val 售后部人员列表 = mutableListOf(Employee("03售后部周某某", false), Employee("03售后部赵某某", true)) val 研发部 = Department("01研发部", 研发部人员列表) val 销售部 = Department("02销售部", 销售部人员列表) val 售后部 = Department("03售后部", 售后部人员列表) val company = mutableListOf(研发部, 销售部, 售后部) // 对每一个部门的人员列表进行排序,在线的排前面,离线的排后面 company.forEach { department -> department.employeeList.sortByDescending { it.isOnline } } // 对部门进行排序,有在线状态的部门排前面 company.sortByDescending { department -> department.employeeList[0].isOnline } company.forEach { println(it) } }

输出结果如下:

Department(name=02销售部, employeeList=[Employee(name=02销售部陈某某, isOnline=true), Employee(name=02销售部李某某, isOnline=false)]) Department(name=03售后部, employeeList=[Employee(name=03售后部赵某某, isOnline=true), Employee(name=03售后部周某某, isOnline=false)]) Department(name=01研发部, employeeList=[Employee(name=01研发部张某某, isOnline=false), Employee(name=01研发部王某某, isOnline=false)])

对于排序,还可以加入更多的逻辑,比如在线状态相同的员工,使用员工名称排序,在线状态相同的部门,使用部门名称排序,假设我们要实现状态相同时,使用名称的降序排序,实现代码如下:

company.forEach { department -> department.employeeList.sortWith(Comparator { employee1, employee2 -> if (employee1.isOnline != employee2.isOnline) { // 状态不相同,则按状态降序排序(即在线的排前面) employee2.isOnline.compareTo(employee1.isOnline) } else { // 状态相同,则按名称升序排序 employee1.name.compareTo(employee2.name) } }) } // 对部门进行排序,有在线状态的部门排前面 company.sortWith(kotlin.Comparator { department1, department2 -> if (department1.employeeList[0].isOnline != department2.employeeList[0].isOnline) { // 部门第一个员工的状态不相同,则按状态降序排序(即有在线员工的部门排前面) department2.employeeList[0].isOnline.compareTo(department1.employeeList[0].isOnline) } else { // 状态相同,则按部门名称升序排序 department1.name.compareTo(department2.name) } })

输出结果如下:

Department(name=02销售部, employeeList=[Employee(name=02销售部陈某某, isOnline=true), Employee(name=02销售部李某某, isOnline=false)]) Department(name=03售后部, employeeList=[Employee(name=03售后部赵某某, isOnline=true), Employee(name=03售后部周某某, isOnline=false)]) Department(name=01研发部, employeeList=[Employee(name=01研发部张某某, isOnline=false), Employee(name=01研发部王某某, isOnline=false)])

对于List的更多详细排序入门,可查看此文章:https://blog.csdn.net/android_cai_niao/article/details/108407853

最新回复(0)