Skip to content

集合特有操作

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