Kotlin之集合排序(sortBy、sortByDescending)

0. 写在前边

最近在工作间隙，试了一把Kotlin。一发车就根本停不下来，各种语法糖，各种炫酷的写法…
闲话少叙，分享一下学习中遇到的集合排序问题

image

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)

--- the end ---

参考：Kotlin 集合排序

最后编辑于：2022.01.17 16:32:18