集合特有操作
Kotlin 集合包包含了集合上的常用扩展函数,例如查找交集、合并或从彼此中减去集合。
要将两个集合合并为一个,请使用 union()
函数。它可以以中缀形式 a union b
使用。请注意,对于有序集合,操作数的顺序很重要。在结果集合中,第一个操作数的元素会排在第二个操作数之前:
kotlin
fun main() {
val numbers = setOf("one", "two", "three")
// output according to the order
println(numbers union setOf("four", "five"))
// [one, two, three, four, five]
println(setOf("four", "five") union numbers)
// [four, five, one, two, three]
}
要查找两个集合之间的交集(同时存在于两个集合中的元素),请使用 intersect()
函数。要查找某个集合中不存在于另一个集合的元素,请使用 subtract()
函数。这两个函数也可以以中缀形式调用,例如 a intersect b
:
kotlin
fun main() {
val numbers = setOf("one", "two", "three")
// same output
println(numbers intersect setOf("two", "one"))
// [one, two]
println(numbers subtract setOf("three", "four"))
// [one, two]
println(numbers subtract setOf("four", "three"))
// [one, two]
}
要查找同时存在于两个集合之一但不在其交集中的元素,你也可以使用 union()
函数。对于此操作(称为对称差集 (symmetric difference)),请计算两个集合之间的差集并合并结果:
kotlin
fun main() {
val numbers = setOf("one", "two", "three")
val numbers2 = setOf("three", "four")
// merge differences
println((numbers - numbers2) union (numbers2 - numbers))
// [one, two, four]
}
你也可以将 union()
、intersect()
和 subtract()
函数应用于列表。然而,它们的返回值_始终_是一个 Set
。在此结果中,所有重复元素都合并为一个,并且不支持索引访问:
kotlin
fun main() {
val list1 = listOf(1, 1, 2, 3, 5, 8, -1)
val list2 = listOf(1, 1, 2, 2, 3, 5)
// result of intersecting two lists is a Set
println(list1 intersect list2)
// [1, 2, 3, 5]
// equal elements are merged into one
println(list1 union list2)
// [1, 2, 3, 5, 8, -1]
}