0. 写在前边
最近在工作间隙,试了一把Kotlin。一发车就根本停不下来,各种语法糖,各种炫酷的写法…
闲话少叙,分享一下学习中遇到的集合排序问题
1. 先构造一个业务类Person 及 Person集合
//定义一个Person类,有name 和 age 两属性
data class Person(var name: String, var age: Int)
var personList: MutableList<Person> = mutableListOf()
personList.add(Person("Jim", 12))
personList.add(Person("A-Lin", 12))
personList.add(Person("Tom", 11))
personList.add(Person("Mary", 14))
2. 使用 sortBy 排序
fun main(args: Array<String>) {
//构造personlist... println("----排序前----")
personList.forEach(::println) println("----排序后----")
//按年龄排序
personList.sortBy ({ it.age })
//或者写成 //personList.sortBy(Person::age)
personList.forEach(::println)
}
运行结果如下:
----排序前----
Person(name=Jim, age=12)
Person(name=A-Lin, age=12)
Person(name=Tom, age=11)
Person(name=Mary, age=14)
----排序后----
Person(name=Tom, age=11)
Person(name=Jim, age=12)
Person(name=A-Lin, age=12)
Person(name=Mary, age=14)
sortBy 为正序排列,跟其对应的降序方法为 sortByDescending
personList.sortByDescending({ it.age })
//运行的结果为
----排序后----
Person(name=Mary, age=14)
Person(name=Jim, age=12)
Person(name=A-Lin, age=12)
Person(name=Tom, age=11)
3. 使用 sortWith 排序
实际情况下,我们的业务需求往往需要根据多个条件来排序,这时我们就需要用到 sortWith 方法
//先根据age 升序排列,若age相同,根据name升序排列
personList.sortWith(compareBy({ it.age }, { it.name }))
// 运行结果如下:
----排序后----
Person(name=Tom, age=11)
Person(name=A-Lin, age=12)
Person(name=Jim, age=12)
Person(name=Mary, age=14)
sortWith 方法中,传入的 compareBy({属性1},{属性2},…)参数, compareBy() 这里的参数个数是可变的,但是都是默认的升序排列。所以我们还可以根据自己的实际需求,给sortWith传入一个Comparator对象,来达到一个更高级更复杂的逻辑
//构造一个Comparator对象,完成排序逻辑:先按age降序排列,若age相同,则按name升序排列
val c1: Comparator<Person> = Comparator { o1, o2 ->
if (o2.age == o1.age) {
o1.name.compareTo(o2.name)
} else {
o2.age - o1.age
}
}
personList.sortWith(c1)
// 运行结果如下:
----排序后----
Person(name=Mary, age=14)
Person(name=A-Lin, age=12)
Person(name=Jim, age=12)
Person(name=Tom, age=11)
4. 通过 data class 实现 Comparable 接口来排序
重新构造 Person 类
data class Person(var name: String, var age: Int) : Comparable<Person> {
override fun compareTo(other: Person): Int {
if (this.age == other.age) {
return this.name.compareTo(other.name)
} else {
return other.age - this.age
}
}
}
调用 list.sorted()方法排序
//sorted 方法返回排序好的list
val sorted = personList.sorted()
sorted.forEach(::println)
//运行结果如下:
----排序后----
Person(name=Mary, age=14)
Person(name=A-Lin, age=12)
Person(name=Jim, age=12)
Person(name=Tom, age=11)