kotlin集合操作——list set map

简述

　　kotlin有三个集合类型——list set map，kotlin标准库提供了一整套用于管理集合的工具

　　这些集合接口与相关函数位于 kotlin.collections 包中

　　kotlin的集合分为可变集合类Mutable和不可变集合类Immutable，List、Set、Map都是不可变集合，与之对应MutableList、MutableSet、MutableMap是可变集合

创建集合

　　使用 listOf()、setOf()、mapOf() 创建不可变的 List、Set、Map

　　使用 mutableListOf()、mutableSetOf()、mutableMap() 创建可变的 MutableList 、MutableSet 、MutableMap

　　注意这里的to代表连接一个键值对，to前面是键，后面是值，set的默认实现是 LinkedHashSet

val list = listOf(1, 2, 3, 4, 5) 
val mutableList = mutableListOf("a", "b", "c")  

val set = setOf(1, 2, 3, 4, 5)
val mutableSet = mutableSetOf("a", "b", "c") 

val map = mapOf(1 to "a", 2 to "b", 3 to "c")
val mutableMap = mutableMapOf(1 to "X", 2 to "Y", 3 to "Z")

遍历集合

　　我们可以使用forEach遍历集合，其中map的遍历可通过it.key和it.value来获取键和值

list.forEach {
     println(it)
}

set.forEach {
     println(it)
}

map.forEach {
     println("K = ${it.key}, V = ${it.value}")
}

　　若想在遍历的同时获取当前元素的下标，我们可以使用forEachIndexed方法

list.forEachIndexed { index, value -> 
     println("list index = ${index}, value = $value")
}

set.forEachIndexed { index, value ->
    println("set index = ${index}, value = ${value}")
}

添加元素

　　对于list，我们可以使用add方法添加元素，参数为一个时往list的末尾添加，两个参数时为往指定的index添加

val list = mutableListOf<Number>(1,2,3)
list.add(4)
list.forEach {
    print("$it ") // 1 2 3 4
}
println()
list.add(3,-1)
list.forEach {
    print("$it ") // 1 2 3 -1 4
}

　　对于set，使用add方法添加元素，注意set是去重的

val set = mutableSetOf<Number>(1,2,3)
set.add(4)
set.forEach{
    println(it)
}

　　对于map，可以使用put方法添加，更常见的是使用中括号+等于的形式，括号内是键名，等号是值

val map = mutableMapOf<Number,String>(1 to "a",2 to "b")
//map.put(3,"c")
map[3]="c"
map.forEach{
    println("${it.key}  ${it.value}")
}

删除元素

　　list使用remove(value)删除值为value的元素，使用removeAt(index)删除指定下标元素

　　removeFirst()和removeLast()删除第一个和最后一个元素

val list = mutableListOf<Number>(10, 20, 30, 40, 50, 60, 70)
list.remove(10) //  20, 30, 40, 50, 60, 70
list.removeAt(2) //  20, 30, 50, 60, 70
list.removeFirst() // 30, 50, 60, 70
list.removeLast() // 30, 50, 60

　　set的remove删除指定值的元素

val set = mutableSetOf<Number>(10, 20, 30, 40)
set.remove(20) // 10 30 40

　　map使用remove删除元素，remove传入想删除的键值

val map = mutableMapOf<Number,String>(1 to "a",2 to "b")
map.remove(2)
map.forEach{
    println("${it.key} ${it.value}") // 1 a
}

list集合相关操作

按索引取元素

　　list本质就可以理解为一个变长数组，最简单的就是像数组一样通过[]拿到值

val numbers = listOf(1, 2, 3, 4)
println(numbers.get(0))
println(numbers[0])
//numbers.get(5)                         // exception!
println(numbers.getOrNull(5))             // null
println(numbers.getOrElse(5, {it}))        // 5

取列表的一部分

　　使用subList可以指定左闭右开的区间，取出list的一部分

val numbers = (0..13).toList()
println(numbers.subList(3, 6))

排序

　　对于普通的基本类型，可以直接调用sort()和sortDescending()进行升序和降序排列

val list = mutableListOf(5,4,1,2,3,7,10)
list.sort() //按升序排序
list.sortDescending() //降序

　　sort()和sortDescending()加上ed变成sorted()和sortedDescending()则是返回一个全新的排序后的list，原来的list不变

　　我们还可以使用sortBy()方法进行排序，例如这里我们对Student类的age属性进行生序排序

　　sortBy是以值的自然顺序进行排序，Int、Double、String等都都定义好的自然顺序

data class Student(
    val age: Int,
    val name: String,
)

fun main() {
    val students = mutableListOf<Student>(
        Student(12, "c"),
        Student(19, "b"),
        Student(15, "a")
    )
    students.sortBy { x -> x.age }
    students.forEach(::println)
}

　　同理sortByDescending是根据降序的自然顺序排序

sortWith

　　查看sortBy的源码我们可以看到内部调用的是sortWith，sortWith常用于根据不同条件进行排序

　　sortWith需要传入一个比较器Comparator参数，我们可以用compareBy生成自己想要的比较器

　　compareBy里面靠左的优先级高，默认是升序排序，例如这里我们先通过age升序排序，age相同时根据name升序排序

data class Student(
    val age: Int,
    val name: String,
)

fun main() {
    val students = mutableListOf<Student>(
        Student(12, "c"),
        Student(19, "b"),
        Student(19, "a"),
        Student(15, "a"),
    )
    students.sortWith(compareBy({ it.age }, { it.name }))
    students.forEach(::println)
}

　　如果需要先a属性升序，再b属性降序排序，则在生成的comparator后使用thenByDescending指定属性进行降序排序

data class Student(
    val age: Int,
    val name: String,
)

fun main() {
    val students = mutableListOf<Student>(
        Student(12, "c"),
        Student(19, "b"),
        Student(19, "a"),
        Student(15, "a"),
    )
    students.sortWith(compareBy<Student> { it.age } .thenByDescending { it.name })
    students.forEach(::println)
}

　　同理可以无限套娃下去，这里的例子是先根据age升序，再根据name降序，再根据bAge升序

data class Student(
    val age: Int,
    val name: String,
    val bAge: Int,
)

fun main() {
    val students = mutableListOf<Student>(
        Student(12, "c",5),
        Student(19, "b",3),
        Student(19, "a",4),
        Student(19, "a",2),
        Student(15, "a",2),
    )
    students.sortWith(compareBy<Student> { it.age }.thenByDescending { it.name }.thenBy { it.bAge })
    students.forEach(::println)
}

GroupBy转map

　　对于一个对象list，我们可以根据对象的属性将list分组，和数据库的groupBy类似

　　例如我们这里有一个student的list

val students = mutableListOf<Student>(
    Student(1, "xiaoming", "guangzhou"),
    Student(2, "xiaohong", "shenzhen"),
    Student(2, "xiaohong2", "shenzhen"),
    Student(2, "xiaohong3", "chongqing"),
    Student(3, "xiaohei", "xian"),
    Student(3, "xiaohei2", "shenzhen"),
)

　　然后我们可以通过student的age进行分组，分组后list会变成一个map，key是我们分组的属性，value是一个list，装着所有符合该分组条件的元素

val afterGroupBy = students.groupBy { it.age }
afterGroupBy.forEach { println("${it.key} ${it.value}") }

　　groupBy里实际上可以放任何数据类型，它的目的就是将list分组转化成map

　　例如下面我们以一个age to name的pair作为groupBy的条件

 

val students = mutableListOf<Student>(
    Student(1, "xiaoming", "guangzhou"),
    Student(2, "xiaohong2", "shenzhen"),
    Student(2, "xiaohong3", "chongqing"),
    Student(2, "xiaohong3", "chongqing"),
    Student(3, "xiaohei", "xian"),
    Student(3, "xiaohei2", "shenzhen"),
)
val afterGroupBy = students.groupBy { it.age to it.name }
afterGroupBy.forEach { println("${it.key} ${it.value}") }

 

　　我们可以看到age为2，name为xiaohong3到student被groupBy起来了

 

map集合相关操作

mapValues

　　对于一个map，我们可以将它的value进行一个映射

　　例如这里有一个有一个Int到String的map，我们对这个map进行mapValues的操作

 

val map = mutableMapOf<Int,String>(1 to "a",2 to "b")
val map2 = map.mapValues { entry -> entry.value + "abc" }

 

　　操作完map2里的value就是map里value拼接上abc的值

 

References

https://www.kotlincn.net/docs/reference/collections-overview.html

https://juejin.cn/post/6957922140767125517

https://www.kotlincn.net/docs/reference/collection-ordering.html

https://developer.android.com/codelabs/basic-android-kotlin-compose-collections?hl=zh-tw#0