Kotlin中的Map排序

时间：02-05来源：作者：点击数：38

示例代码如下：

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)])